BZOJ 1088 [SCOI2005]扫雷Mine

【题解】

　　设给出的信息为$a[i]$，每个格子的雷数为$f[i]$

　　我们可以发现$f[i]=a[i-1]-f[i-1]-f[i-2]$

　　也就是说只要确定了前两个格子的情况，全部的情况都可以计算了。

　　那么我们枚举前两个格子的情况，检查是否合法即可。

　　

#include<cstdio>
#include<algorithm>
#define N 200010
#define rg register
using namespace std;
int n,m,a[N],f[N],ans; 
inline int read(){
	int k=0,f=1; char c=getchar();
	while(c<'0'||c>'9')c=='-'&&(f=-1),c=getchar();
	while('0'<=c&&c<='9')k=k*10+c-'0',c=getchar();
	return k*f;
}
inline int check(){
	for(rg int i=3;i<=n;i++){
		f[i]=a[i-1]-f[i-2]-f[i-1];
		if(f[i]<0) return 0;	
	}
	if(f[n-1]+f[n]!=a[n]) return 0;
	return 1;
}
int main(){
	n=read();
	for(int i=1;i<=n;i++) a[i]=read();
	if(a[1]==0){
		f[1]=f[2]=0; ans+=check();
	}
	if(a[1]==1){
		f[1]=0; f[2]=1; ans+=check();
		f[1]=1; f[2]=0; ans+=check();
	}
	if(a[1]==2){
		f[1]=f[2]=1; ans+=check();
	}
	return printf("%d\n",ans),0;
}