扫雷 HDU - 5965

扫雷 HDU - 5965

Problem Description
扫雷游戏是晨晨和小璐特别喜欢的智力游戏，她俩最近沉迷其中无法自拔。
该游戏的界面是一个矩阵，矩阵中有些格子中有一个地雷，其余格子中没有地雷。 游戏中，格子可能处于己知和未知的状态。如果一个己知的格子中没有地雷，那么该 格子上会写有一个一位数，表示与这个格子八连通相邻的格子中地雷总的数量。
现在，晨晨和小璐在一个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

Sample Output
6
1

思路
模拟+枚举+dp
分析可得，如果我们知道了第一列的雷的个数，则后面每列的雷的个数可以确定，假设num[i]表示第i列题目告诉我们的雷的个数，dp[i]为第i列所放的雷的个数，则dp[i]=num[i]-dp[i-1]-dp[i-2]; 而第一列的雷的个数有三种情况 0，1，2.枚举即可。
每次判断dp[i]是否满足>=0&&<=2.最后再判断最后的两列是否满足dp[l]+dp[i-1]==num[l];
设每次枚举的方案数ans=1 如果dp[i]=0或2 则ans乘1，若为1，则ans乘2. 再相加 sum+=ans.
下面是代码

#include <iostream> 
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <algorithm>
#include <climits>
#include <cstring>
#include<sstream>
#include <string>
#include <set>
#include <map>
#include <queue>
#include <stack>
#include <vector>
#include <list>
#define rep(i,m,n) for(i=m;i<=n;i++)
#define rsp(it,s) for(set<int>::iterator it=s.begin();it!=s.end();it++)
const int Mod = 1e8 + 7;  
using namespace std;
int dp[10005],num[10005];
char s[10005];
long long int sum,ans;
int main()
{
    int T,l;
    cin>>T;
    getchar();
    while(T--)
    {
        memset(dp,0,sizeof(dp));
        sum=0;                 //初始化
        gets(s);
        l=strlen(s);
        for(int i=0;i<l;i++)
        num[i+1]=s[i]-'0';         //将字符串转化为数字
        for(int i=0;i<=num[1];i++)   //枚举位置1上的情况
        {
            int flag=1;
            ans=1;
            if(i>2) break;         //可能的情况0，1，2
            dp[1]=i;
            int j;
            for(j=2;j<=l;j++)
            {

                dp[j]=num[j-1]-dp[j-1]-dp[j-2];
                if(dp[j]>2||dp[j]<0)    //可能的情况0，1，2
                {
                    flag=0;
                    break;
                }

            }
           if(j==l+1&&dp[l]+dp[l-1]!=num[l]) 
           flag=0;
           //判断最后两列是否满足要求，（这里的判断不要放到前面的for循环里，我前面的循环从第二列开始，而它可能只有一列，这里的bug我找了半天。。。）
           if(flag==1)
           {
            for(j=1;j<=l;j++)
            {
                if(dp[j]==1) ans=ans*2%Mod; //如果是0或2则只有一种可能，如果是1则有两种
            }
            sum=(sum+ans)%Mod;  //将每次枚举得出的可能加到sum中
          }
        }
      cout<<sum<<endl;    //输出结果
    }
    return 0;
}