P2327 [SCOI2005]扫雷

P2327 [SCOI2005]扫雷

题目描述

输入输出格式

输入格式：

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

输出格式：

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

输入输出样例

输入样例#1：

2
1  1

输出样例#1：

2

分析：只有两种情况，第一个是雷，或者第二个是雷，试一下他们对不对即可

#include<cstdio>
#define MAXN 10100

int f[MAXN],a[MAXN];
int n,ans = 0;
bool dp(int x)
{
    f[1] = x;
    for (int i=2; i<=n; ++i)
    {
        f[i] = a[i-1]-f[i-1]-f[i-2];
        if (f[i]<0||f[i]>1) return false;
    }
    if (f[n]+f[n-1]!=a[n]) return false ;
    return true ;
}
int main()
{
    scanf("%d",&n);
    for (int i=1; i<=n; ++i)
        scanf("%d",&a[i]);
    if (dp(0)) ans++;
    if (dp(1)) ans++;
    printf("%d",ans);
    return 0;
}