hdu 5965 扫雷

题意：三行n列的棋盘，给出第二列的每个格子的雷数a[i]（代表周围八个格子的地雷数量，中间那一行肯定没地雷），问有多少种放雷方案

思路：每一列只能放0，1，2，我们对于第一列进行枚举，dp[i]代表每一列雷数目,所以dp[2]=a[1]-dp[1],dp[3]=a[2]-dp[1]-dp[2]...所以每一列是确定的

 

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll mod=1e8+7;

int a[10004];
int dp[10004];
int n;

ll hh(int x){
    memset(dp,0,sizeof(dp));
    dp[1]=x;
    dp[2]=a[1]-dp[1];
    if(dp[2]<0||dp[2]>2) return 0;
    for(int i=3;i<=n;i++){
        dp[i]=a[i-1]-dp[i-1]-dp[i-2];
        if(dp[i]<0||dp[i]>2) return 0;
    }
    if((dp[n-1]+dp[n])!=a[n]) return 0;
    ll s=1;
    for(int i=1;i<=n;i++){
        if(dp[i]==1) s=s*2%mod;
    }
    return s;
}
char s[10002];
int main(){
    int t;
    cin>>t;
    while(t--){
        scanf("%s",s+1);
        n=strlen(s+1);
        for(int i=1;i<=n;i++) {
            a[i]=s[i]-'0';
        }
        ll sum=0;
        for(int i=0;i<3;i++)
            sum=(sum+hh(i))%mod;
        printf("%lld\n",sum);
    }
    return 0;
}