BZOJ 1088 扫雷MIne 枚举

题意:n*2棋盘,第一列里面某些格子是雷，而第二列没有雷.
如果某个格子没有雷，那么它里面的数字表示和它8连通的格子里面雷的数目.
n<=1e4,现在给出第二列上的数,求出第一列有多少种摆雷方案数?


刚开始还想着保存前两行状态然后DP ...完全没必要.

如果a[i-2]和a[i-1]确定以后,a[i]受b[i-1]限制也确定下来.

则枚举前两个元素后,其余元素也就确定下来了.

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=2e4+20;
const ll mod=1e9+7;
int n,b[N],a[N]; 
int main()
{ 
	int res=0;
	cin>>n;
	for(int i=1;i<=n;i++)
		scanf("%d",&b[i]);
	if(n==1)
		puts("1");
	else
	for(int i=0;i<4;i++)
	{
		a[1]=(i>>1)&1,a[2]=i&1;
		bool flag=true;
		if(a[1]+a[2]!=b[1])
			continue;
		for(int j=3;j<=n;j++)
		{
			a[j]=b[j-1]-a[j-2]-a[j-1];
			if(a[j]<0)
				flag=false;
		}	
		if(a[n-1]+a[n]!=b[n])
			flag=false;
		if(flag)
			res++;
	}
	cout<<res<<endl;
	return 0;
}