【BZOJ 1088】【SCOI 2005】扫雷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>#include<algorithm>using namespace std;#define N 10010int a[N],f[N];int n,ans;bool jud(){    for(int i = 2;i <= n;i++)    {        f[i+1] = a[i] - f[i] - f[i-1];        if(f[i+1] < 0) return 0;    }    if(a[n] != f[n] + f[n-1]) return 0;    return 1;}int main(){    scanf("%d",&n);    for(int i = 1;i <= n;i++) scanf("%d",&a[i]);    ans = 0;    if(a[1] == 0) ans += jud();    else        if(a[1] == 1)//分类讨论         {            f[1] = 1; ans += jud();            memset(f,0,sizeof(f));            f[2] = 1; ans += jud();        }        else        { f[1] = f[2] = 1;ans += jud(); }    printf("%d",ans);    return 0;}