hdu 5965 扫雷

扫雷

Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 65536/65536 K (Java/Others)
Total Submission(s): 350    Accepted Submission(s): 72

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
  
  

 

Source

2016年中国大学生程序设计竞赛（合肥）-重现赛（感谢安徽大学）

 

明显的dp题目，把三行变两行，对于有雷的那行，对于每一个格，如果有2个或一个雷，那么有一种方法，如果有1个雷，就有两种方法（可以上面一个或者下面一个），我们定义b[i] 为i位置上的雷，那么b[i]=a[i-1]-b[i-1]-b[i-2]。a[i]为第二行第i位置上的数字。一直模拟下去，如果最后a[len-1]!=b[len-1]+b[len-2]则不可能发生这种情况。其中len为矩阵长度，数组下标从0开始。

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<vector>
#include<queue>
#include<string.h>
using namespace std;
const int mod=1e8+7;
const int maxn=10005;
char pp[10005];
int p[maxn];
int dp[maxn][15][15];
int b[maxn];
int main()
{
    int cas;
    scanf("%d",&cas);
    while(cas--)
    {
       scanf("%s",&pp);
        //cin>>pp;
        int len=strlen(pp);
        for(int i=0;i<len;i++)
        {
            p[i]=pp[i]-'0';
        }    
         if(len==1)
        {
            if(p[0]==1) puts("2");
            else if(p[0]==0||p[0]==2) puts("1");
            else puts("0");
            continue;
        }
         long long sum=0;
        for(int i=0;i<=min(2,p[0]);i++)
        {
            if(p[0]-i>2||p[0]-i<0)
                continue;
            b[1]=i;b[0]=p[0]-b[1];
            long long ans=1;
            if(b[0]==1)
                ans=(ans*2)%mod;
            else if(b[0]==2||b[0]==0)
                ans=ans*1%mod;
            if(b[1]==1)
                ans=(ans*2)%mod;
            else if(b[1]==0||b[1]==2)
                ans=(ans*1)%mod;
            for(int j=2;j<len;j++)
            {
                b[j]=p[j-1]-b[j-1]-b[j-2];
                if(b[j]<0||b[j]>2)
                {
                    ans=0;
                    break;
                }
                if(b[j]==0||b[j]==2)
                    ans=(ans*1)%mod;
                else if(b[j]==1)
                    ans=(ans*2)%mod;
            }
            if(b[len-1]+b[len-2]!=p[len-1])
              ans=0;
            sum=(sum+ans)%mod;
        }
        printf("%lld\n",sum);
    }
}