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

题解

暴力判断是否合法即可。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<algorithm>
using namespace std;
int n,ans;
int a[10002],f[10001];
int check()
{
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
-f[n-1]-f
!=0) return 0;//一定要写这句话
return 1;
}
void init()
{
scanf("%d",&n);
for(int i=1;i<=n;i++) scanf("%d",&a[i]);
}
int main()
{
init();
if(a[1]==0) ans+=check();
else if(a[1]==1)
{f[1]=1; ans+=check();
memset(f,0,sizeof(f));
f[2]=1; ans+=check();
}
else {f[1]=f[2]=1; ans+=check();}
printf("%d",ans);
return 0;
}