【SCOI2005】【BZOJ1088】扫雷Mine

Description

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

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

一个数，即第一列中雷的摆放方案数。
Sample Input
2
1 1
Sample Output
2
HINT

Source

n只有2,所以状态数显然是非常少的.
我们枚举第一个位置0/1,然后显然可以直接推出后面所有雷的状态,然后如果不合法的话,n+1位置一定是有雷的.

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#define MAXN 10010
#define GET (ch>='0'&&ch<='9')
using namespace std;
int n;
int a[MAXN],f[MAXN];
int ans;
void in(int &x)
{
    char ch=getchar();x=0;
    while (!GET)    ch=getchar();
    while (GET) x=x*10+ch-'0',ch=getchar();
}
bool check()
{
    for (int i=2;i<=n;i++)  f[i+1]=a[i]-f[i]-f[i-1];
    return !f[n+1];
}
int main()
{
    in(n);
    for (int i=1;i<=n;i++)  in(a[i]);
    for (int i=0;i<=a[1];i++)   memset(f,0,sizeof(f)),f[1]=i,f[2]=a[1]-i,ans+=check();
    cout<<ans<<endl;
}