【bzoj 1088/P2327】扫雷 2005SCOI（脑洞？）

感觉codevs和bzoj上的都不如luogu上的好看23333，毕竟人家有图嘛

Description

　　相信大家都玩过扫雷的游戏。那是在一个n*m的矩阵里面有一些雷，要你根据一些信息找出雷来。万圣节到了，‘’余”人国流行起了一种简单的扫雷游戏，这个游戏规则和扫雷一样，如果某个格子没有雷，那么它里面的数字表示和它8连通的格子里面雷的数目。现在棋盘是n×2的，第一列里面某些格子是雷，而第二列没有雷，如下图：

　　？？？？？？？？？？？

由于第一列的雷可能有多种方案满足第二列的数的限制，你的任务即根据第二列的信息确定第一列雷有多少种摆放方案。

Input

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

Output

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

Sample Input

2

1 1

Sample Output

2

思路

由某个点的前一点的数字和前两点的雷可以确定这个点是否防雷

只要第一个雷已经确定，那么后面所有的雷都是确定的，

用 a 表示第二列的数字；

用 b 表示第一列的雷，若为1则有雷，反之亦然

所以枚举b[1]

接下来 b[i] = a[i - 1] - b[i -1] - b[i - 2]

判断中间的情况是否合法

最后记得判断最后一个是否合法。

如果最后b
+b[n-1]<>a
，那么就能说明这种方案出错了，不进行ans++

代码如下：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
typedef long long LL;
#define ri register int
const int sz = 100010;
inline void read(int &x){
x=0;bool fl=0;char c=getchar();
while(c<'0'||c>'9')
{if(c=='-') fl=1;c=getchar();}
while(c>='0
4000
'&&c<='9')
{x=x*10+c-'0';c=getchar();}
if(fl) x*=-1;
}
inline void write(int x){
if(x<0) putchar('-'),x*=-1;
if(x/10) write(x/10);
putchar(x%10+'0');
}
int n,a[sz],b[sz],ans;
inline bool check()
{
for(ri i=1;i<=n;++i)
{
b[i+1]=a[i]-b[i]-b[i-1];
if(b[i+1]<0||b[i+1]>1)
return 0;
}
if(b[n+1]!=0) return 0;
return 1;
}
int main()
{
read(n);
for(ri i=1;i<=n;++i)    read(a[i]);
if(a[1]==0)
ans+=check();
else if(a[1]==1)
{
b[1]=1;
ans+=check();
memset(b,0,sizeof(b));
b[2]=1;
ans+=check();
}
else
{
b[1]=1;
b[2]=1;
ans+=check();
}
write(ans),puts("");
return 0;
}