P2327 [SCOI2005]扫雷

题目描述

输入输出格式

输入格式：

第一行为N，第二行有N个数，依次为第二列的格子中的数。（1<= N <= 10000）

输出格式：

一个数，即第一列中雷的摆放方案数。

输入输出样例

输入样例#1：

2
1  1

输出样例#1：

2

 ‘

我们用dp[i]表示左边的第i个位置是否有雷，

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<queue>
#include<algorithm>
#include<cstdlib>
using namespace std;
const int MAXN=10001;
void read(int &n)
{
    char c='+';int x=0,flag=1;
    while(c<'0'||c>'9')
    {c=getchar();if(c=='-')flag=-1;}
    while(c>='0'&&c<='9')
    {x=x*10+c-48;c=getchar();}
    n=(x*flag);
}
int k=1,m,n;
int i,j;
int a[MAXN],dp[MAXN];
int main()
{
    read(n);
    for(i=1;i<=n;i++)
        read(a[i]);
    for(i=2;i<=n+1;i++)
    {
        dp[i]=a[i-1]-dp[i-1]-dp[i-2];
        if(dp[i]<0||dp[i]>1)
        {
            k=0;
            break;
        }
    }
    if(dp[n+1]!=0)k=0;
    m+=k;
    memset(dp,0,sizeof(dp));
    k=1;
    dp[1]=1;
    for(i=2;i<=n+1;i++)
    {
        dp[i]=a[i-1]-dp[i-1]-dp[i-2];
        if(dp[i]<0||dp[i]>1)
        {
            k=0;
            break;
        }
    }
    if(dp[n+1]!=0)k=0;
    m+=k;
    printf("%d
",m);
    return 0;
}