Description
相信大家都玩过扫雷的游戏。那是在一个n*m的矩阵里面有一些雷,要你根据一些信息找出雷来。万圣节到了,“余”人国流行起了一种简单的扫雷游戏,这个游戏规则和扫雷一样,如果某个格子没有雷,那么它里面的数字表示和它8连通的格子里面雷的数目。现在棋盘是n×2的,第一列里面某些格子是雷,而第二列没有雷,如下图:由于第一列的雷可能有多种方案满足第二列的数的限制,你的任务即根据第二列的信息确定第一列雷有多少种摆放方案。
Input
第一行为N,第二行有N个数,依次为第二列的格子中的数。(1<= N <= 10000)
Output
一个数,即第一列中雷的摆放方案数。
Sample Input
2
1 1
Sample Output
2
我们设a[i]表示i点周围点的雷数,b[i]表示i点是否有雷,可以推出
a[i]=b[i-1]+b[i]+b[i+1];
所以b[i]=a[i-1]-b[i-1]-b[i-2];
所以枚举b[1],可以递推求出方案数
code:
#include<bits/stdc++.h>
using namespace std;
int i,j,k,n,m,re,tot,ans;
int a[10005],b[10005];
int read(){
char c;int x;while(c=getchar(),c<'0'||c>'9');x=c-'0';
while(c=getchar(),c>='0'&&c<='9') x=x*10+c-'0';return x;
}
int main()
{
n=read();
for(int i=1;i<=n;i++) a[i]=read();
for(int i=0;i<=1;i++){
b[1]=i;ans+=1;re=0;
for(int j=2;j<=n;j++){
b[j]=a[j-1]-b[j-1]-b[j-2];
if(b[j]>1||b[j]<0){
re=1;ans--;break;
}
}
if(re==1) continue;
if(b[n]+b[n-1]!=a[n]) ans--;
}
printf("%d",ans);
return 0;
}