[bzoj1088][SCOI2005]扫雷Mine

1088: [SCOI2005]扫雷Mine

Time Limit: 10 Sec   Memory Limit: 162 MB
Submit: 3706   Solved: 2182
[ Submit][ Status][ Discuss]

Description

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

Input

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

Output

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

Sample Input

2
1 1

Sample Output

2

#include<cstdio>
int n,ans=0;
int a[10001],b[10001];
int plus(){
	for( int i = 2; i < n; i++ ) { b[i+1] = a[i]-b[i]-b[i-1]; if(b[i+1]<0) return 0;}
	if( a[n]-b[n]-b[n-1]==0 ) return 1;
	return 0;
}
template<class T>inline void read(T &res){
	static char ch;T flag=1;
	while((ch=getchar())>'9'||ch<'0')if(ch=='-')flag=-1;res=ch-48;
	while((ch=getchar())>='0'&&ch<='9')res=res*10+ch-48;res*=flag;
}
int main(){
	read(n);
	for( int i = 1; i <= n; i++ ) read(a[i]);
	if( a[1] == 0 ) ans += plus();
	else if( a[1] == 1 ){
		b[1] = 1; ans += plus();
		b[1] = 0; b[2] = 1; ans += plus();
	} else{ b[1] = b[2] = 1; ans += plus(); }
	printf("%d",ans);
	return 0;
}