bzoj 1088 [SCOI2005]扫雷Mine

Description

　　相信大家都玩过扫雷的游戏。那是在一个n*m的矩阵里面有一些雷，要你根据一些信息找出雷来。万圣节到了
，“余”人国流行起了一种简单的扫雷游戏，这个游戏规则和扫雷一样，如果某个格子没有雷，那么它里面的数字
表示和它8连通的格子里面雷的数目。现在棋盘是n×2的，第一列里面某些格子是雷，而第二列没有雷，如下图： 
由于第一列的雷可能有多种方案满足第二列的数的限制，你的任务即根据第二列的信息确定第一列雷有多少种摆放
方案。

Input

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

Output

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

Sample Input

2
1 1

Sample Output

2

只要知道第一个格子有无雷，根据a[1]，可得到第二个格子有无雷，根据a[3]，知道第三个格子有无雷。。。。依次类推，看是否有矛盾。

答案只能是0、1、2。

#pragma GCC optimize(2)
#include <bits/stdc++.h>
#define ll long long
using namespace std;

const int N = 1e4+10;
int a[N],b[N],c[N],n;
int judge()
{
    for(int i=2;i<=n-1;i++)
    {
        b[i+1]=a[i]-a[i-1]+b[i-2];
        if(b[i+1]<0||b[i+1]>1)
            return 0;
    }
    return b[n]+b[n-1]==a[n];
}
int main(){
//    freopen("2.in","r",stdin);
    cin>>n;
    for(int i=1;i<=n;i++)
    {
        cin>>a[i];
    }
    int ans=0;
    if(a[1]==0)
    {
        b[1]=b[2]=0;
        ans+=judge();
    }
    else if(a[1]==1)
    {
        b[1]=0,b[2]=1;ans+=judge();
        b[1]=1,b[2]=0,ans+=judge();
    }
    else if(a[1]==2)
    {
        b[1]=b[2]=1;ans+=judge();
    }
    cout<<ans<<'\n';
    return 0;
}