[SCOI2005]扫雷Mine

Description

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

Input

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

Output

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

Sample Input

2

1 1
Sample Output

2

#include<cstdio>
#include<cstring>
using namespace std;
int a[11000];
int f[11000],n;
int solve()
{
    for(int i=2;i<n;i++)
    {
        int t=a[i]-f[i]-f[i-1];
        if(t<0 || t>1)return 0;
        else f[i+1]=t;
    }
    if(f[n]+f[n-1]!=a[n])return 0;
    return 1;
}
int main()
{
    int ans=0;
    scanf("%d",&n);
    for(int i=1;i<=n;i++)scanf("%d",&a[i]);
    memset(f,0,sizeof(f));
    if(a[1]>2){printf("0\n");return 0;}
    if(a[n]>2){printf("0\n");return 0;}
    if(a[1]==2)
    {
        f[1]=1;f[2]=1;
        ans+=solve();
    }
    if(a[1]==1)
    {
        memset(f,0,sizeof(f));
        f[1]=1;
        ans+=solve();
        memset(f,0,sizeof(f));
        f[2]=1;
        ans+=solve();
    }
    if(a[1]==0)
    {
        memset(f,0,sizeof(f));ans+=solve();
    }
    printf("%d\n",ans);
    return 0;
}