hdu5965 dp枚举

http://acm.hdu.edu.cn/showproblem.php?pid=5965

用num[i]表示第i个格子里的数，dp[i]表示第i列有多少个雷，由dp[i]=num[i]-dp[i-1]-dp[i-2]，枚举第一个格子里的雷可能的数量

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;

typedef long long ll;
const int maxn=10005;
const int mod=1e8+7;
char s[maxn];
int num[maxn],dp[maxn];
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%s",s);
        int n=strlen(s);
        memset(dp,0,sizeof(dp));
        for(int i=0; i<n; i++)
            num[i+1]=s[i]-'0';
        memset(dp,0,sizeof(dp));
        ll ans=0;
        for(int i=0; i<=num[1]; i++)
        {
            dp[1]=i;
            if(i>2)break;
            int j;
            for(j=2; j<=n; j++)
            {
                int k=num[j-1]-dp[j-1]-dp[j-2];
                if(k>2||k<0)
                    break;
                dp[j]=k;
            }
            if(j<=n)continue;
            if(dp[n-1]+dp[n]!=num[n])continue;
            ll res=1;
            for(int i=1; i<=n; i++)
            {
                if(dp[i]==1)
                    res*=2;
                res%=mod;
            }
            ans+=res;
            ans%=mod;
        }
        printf("%d\n",ans);
    }
    return 0;
}