扫雷
题目
扫雷游戏是晨晨和小璐特别喜欢的智力游戏,她俩最近沉迷其中无法自拔。
该游戏的界面是一个矩阵,矩阵中有些格子中有一个地雷,其余格子中没有地雷。 游戏中,格子可能处于己知和未知的状态。如果一个己知的格子中没有地雷,那么该 格子上会写有一个一位数,表示与这个格子八连通相邻的格子中地雷总的数量。
现在,晨晨和小璐在一个3行N列(均从1开始用连续正整数编号)的矩阵中进 行游戏,在这个矩阵中,第2行的格子全部是己知的,并且其中均没有地雷;而另外 两行中是未知的,并且其中的地雷总数量也是未知的。
晨晨和小璐想知道,第1行和第3行有多少种合法的埋放地雷的方案。
Input
包含多组测试数据,第一行一个正整数T,表示数据组数。
每组数据由一行仅由数字组成的长度为N的非空字符串组成,表示矩阵有3行N 列,字符串的第i个数字字符表示矩阵中第2行第i个格子中的数字。
保证字符串长度N <= 10000,数据组数<= 100。
Output
每行仅一个数字,表示安放地雷的方案数mod100,000,007的结果。
Sample
Input
2
22
000
Output
6
1
解题思路
刚拿到题,一直在画三行的方格,各种举例企图找到点规律,就是先枚举出第一行放雷的情况然后再根据规则判断第三行放雷的情况,可能可以实现但是越写越乱就放弃了。
然后想到可以把第一行和第三行的雷放在一行中,那么总共就只剩两行。第一行表示原来第二行的数字,第二行则表示第一行和第三行的地雷数目,这样我们只要确定第二行第一列的放雷数目后,根据判断规则依次算出第二行第一列后面的地雷数。
如图所示(下标从0开始),设第二行的放雷情况为dp,在确定了dp[0]的地雷数后,就可以通过dp[0]计算出dp[1]的地雷书,从dp[2]开始,第n列的放雷为 d p [ n ] = a [ n − 1 ] − d p [ n − 1 ] − d p [ n − 2 ] dp[n] = a[n-1] - dp[n-1] - dp[n-2] dp[n]=a[n−1]−dp[n−1]−dp[n−2],可以根据这个规律一直算到最后一列。但是我需要根据放雷的规则,判断第一行最后一列a[len-1]下的地雷书是否满足要求,即 d p [ l e n − 1 ] + d p [ l e n − 2 ] = = a [ l e n − 1 ] dp[len - 1] + dp[len - 2] == a[len - 1] dp[len−1]+dp[len−2]==a[len−1],若满足要求,则证明所设定的dp[0]下的地雷数i是正确的,否则是错误的。
需要注意的是,由于原先的第一行和第三行的地雷范围只能是0或者1(不放雷或者放一个雷),所以dp行的取值只能够是0、1和2,也就是原先的第一行和第三行都不放雷、原先的第一行或者第三行只有一行放雷、原先的第一行和第三行都放雷这三种情况。所以可以枚举出dp第一列即dp[0]的取值为0,1,2的情况,最后将这三种情况的总数相加即可。
最后再计算总数时需要注意,dp为1的时候有两种情况,也就是原先的第一行或者第三行只有一行放雷两种。
AC
#include <iostream>
#include <cstring>
#include <cstdio>
using namespace std;
#define mod 100000007
int main()
{
int n,flag = 1;
cin >> n;
while(n--)
{
int sum = 0;
char str[10000];
int dp[10000],a[10000]; //dp为第2行雷的个数。
cin >> str;
int N = strlen(str);
for(int i = 0; i < N; i ++)
a[i] = str[i] - '0';
for(int i = 0; i <= 2 ; i ++) //dp行各列数值范围只能是[0,2]
{
flag = 1;
dp[0] = i; //枚举dp行第一列的数值0,1,2的情况。
dp[1] = a[0] - dp[0]; //计算出dp第二列的值。
if(dp[1] < 0 || dp[1] > 2) //判断是否合法。
flag = 0;
for(int j = 2; j < N ; j ++)
{
dp[j] = a[j-1] - dp[j-1] - dp[j-2];
if(dp[j] < 0 || dp[j] > 2) //判断是否合法。
flag = 0;
}
if(dp[N - 1] + dp[N - 2] != a[N - 1]) //判断最后一列是否合法。
flag = 0;
if(flag)
{
int count = 1;
for(int i = 0 ; i < N ; i ++)
if(dp[i] == 1) //当dp行数值为1时,有两种放雷法,数值为0,2时只有一种。
count = count * 2 % mod;
sum = (sum + count)%mod;
}
}
cout<<sum<<endl;
}
return 0;
}