您的位置:首页 > 其它

hdu5965 dp枚举

2016-11-09 18:16 204 查看
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
!=num
)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;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: