Description
相信大家都玩过扫雷的游戏。那是在一个n*m的矩阵里面有一些雷,要你根据一些信息找出雷来。万圣节到了
,“余”人国流行起了一种简单的扫雷游戏,这个游戏规则和扫雷一样,如果某个格子没有雷,那么它里面的数字
表示和它8连通的格子里面雷的数目。现在棋盘是n×2的,第一列里面某些格子是雷,而第二列没有雷,如下图:
由于第一列的雷可能有多种方案满足第二列的数的限制,你的任务即根据第二列的信息确定第一列雷有多少种摆放
方案。
Input
第一行为N,第二行有N个数,依次为第二列的格子中的数。(1<= N <= 10000)
Output
一个数,即第一列中雷的摆放方案数。
Sample Input
2
1 1
1 1
Sample Output
2
HINT
一个比较简单的dp。
f[i][0/1][0/1]表示前i位,第i位没雷/有雷,第(i-1)没雷/有雷的总方案数。
根据第i位上的数字直接往下推即可。
num=1:
f[i+1][1][0]+=f[i][0][0],
f[i+1][0][0]+=f[i][0][1],
f[i+1][0][1]+=f[i][1][0];
num=2:
f[i+1][0][0]+=f[i][0][1],
f[i+1][0][1]+=f[i][1][0];
num=2:
f[i+1][1][1]+=f[i][1][0],
f[i+1][1][0]+=f[i][0][1],
f[i+1][0][1]+=f[i][1][1];
num=3:
f[i+1][1][0]+=f[i][0][1],
f[i+1][0][1]+=f[i][1][1];
num=3:
f[i+1][1][1]+=f[i][1][1];
初始化f[1][0][0]=f[1][1][0]=1
然后答案就是f[n+1][0][0]+f[n+1][0][1]
不难理解吧。
问题就是……这题竟然会有数字0的情况!!
喂,扫雷的时候数字0的话是个空格好吗,你哪儿来的数字0 = =
所以再加一个num=0的转移吧……
f[i+1][0][0]+=f[i][0][0];
#include<bits/stdc++.h>
using namespace std;
const int
N=10005;
int n,f[N][2][2];
int main(){
scanf("%d",&n);
f[1][0][0]=f[1][1][0]=1;
int num;
for (int i=1;i<=n;i++){
scanf("%d",&num);
if (!num) f[i+1][0][0]+=f[i][0][0];
if (num==1) f[i+1][1][0]+=f[i][0][0],
f[i+1][0][0]+=f[i][0][1],
f[i+1][0][1]+=f[i][1][0];
if (num==2) f[i+1][1][1]+=f[i][1][0],
f[i+1][1][0]+=f[i][0][1],
f[i+1][0][1]+=f[i][1][1];
if (num==3) f[i+1][1][1]+=f[i][1][1];
}
printf("%d\n",f[n+1][0][0]+f[n+1][0][1]);
return 0;
}