题目:
相信大家都玩过扫雷的游戏。那是在一个n*m的矩阵里面有一些雷,要你根据一些信息找出雷来。万圣节到了
,“余”人国流行起了一种简单的扫雷游戏,这个游戏规则和扫雷一样,如果某个格子没有雷,那么它里面的数字
表示和它8连通的格子里面雷的数目。现在棋盘是n×2的,第一列里面某些格子是雷,而第二列没有雷,如下图:
由于第一列的雷可能有多种方案满足第二列的数的限制,你的任务即根据第二列的信息确定第一列雷有多少种摆放
方案。
Input
第一行为N,第二行有N个数,依次为第二列的格子中的数。(1<= N <= 10000)
Output
一个数,即第一列中雷的摆放方案数。
Sample Input
2 1 1
Sample Output
2
思路:
因为一个位置有地雷会牵动下面三个方格,所以仅仅是建二维dp是不对的,需要建立三维dp数组。
第一维表示的是i位置。
第二维表示的是第i个位置放没放地雷,0表示没放,1表示放了。
第三维表示的是第i个位置放没放地雷,0表示没放,1表示放了。
这样就可以把一个地雷影响的方格全部包含了,下面就是推递推方程了。
显然:
第i个位置:
a[i]==0时:
只有000情况。三个数依次是i-1,i,i+1,地雷的摆放情况。
dp[i+1][0][0]=dp[i][0][0];
为啥是从i+1呢,因为只有从i+1开始才能完全包含三个方格而不包含别的方格。
a[i]==1时:
有三种情况:001,010,100,分别对应下面递推方程:
dp[i+1][1][0]=dp[i][0][0];
dp[i+1][0][1]=dp[i][1][0];
dp[i+1][0][0]=dp[i][0][1];
a[i]==2时:
也有三种情况:110,101,011,分别对应下面递推方程:
dp[i+1][0][1]=dp[i][1][1];
dp[i+1][1][0]=dp[i][0][1];
dp[i+1][1][1]=dp[i][1][0];
a[i]==3时,
只有一种情况:111
递推方程为:
dp[i+1][1][1]=dp[i][1][1];
然后只要输出dp[n+1][0][1]+dp[n+1][0][0]就是正确答案.
代码如下:
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
using namespace std;
typedef long long ll;
const int maxn=1e4+5;
int n;
int a[maxn];
int dp[maxn][2][2];
int main()
{
while(scanf("%d",&n)!=EOF)
{
memset (a,0,sizeof(a));
for (int i=1;i<=n;i++) scanf("%d",&a[i]);
memset (dp,0,sizeof(dp));
if(a[1]==0)
{
dp[2][0][0]=1;
}
if(a[1]==1)
{
dp[2][0][1]=1;
dp[2][1][0]=1;
}
if(a[1]==2)
{
dp[2][1][1]=1;
}
for (int i=2;i<=n;i++)
{
if(a[i]==0)
{
dp[i+1][0][0]=dp[i][0][0];
}
if(a[i]==1)
{
dp[i+1][1][0]=dp[i][0][0];
dp[i+1][0][1]=dp[i][1][0];
dp[i+1][0][0]=dp[i][0][1];
}
if(a[i]==2)
{
dp[i+1][0][1]=dp[i][1][1];
dp[i+1][1][0]=dp[i][0][1];
dp[i+1][1][1]=dp[i][1][0];
}
if(a[i]==3)
{
dp[i+1][1][1]=dp[i][1][1];
}
}
printf("%d\n",dp[n+1][0][0]+dp[n+1][0][1]);
}
return 0;
}