题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=5389
题面:
4 3 9 1 1 2 6 3 9 1 2 3 3 5 2 3 1 1 1 1 1 9 9 9 1 2 3 4 5 6 7 8 9
1 0 10 60
开始没想到递推,队友说了下dp,往这方面想了下就想到了。状态转移方程为dp[i][j]=(dp[i-1][tmp]+dp[i-1][j])%mod;其中tmp为j-arr[i],i是当前下标,j从1到9,若算出的tmp小于等于0还需加上9。dp[i][j]的含义是到第i位为止,前i个数中的若干个加起来和为j的方案数,转移方程中的两项分别对应,当前i位取或不取。因为每个数字是由确定的两个数字组合加起来得到的,比如2,若其中一个为1,那么另外一个也一定为1。且存在这样的性质,若干个数,按给定的规则先加,或者分开加,再组合在一起,得到的结果是一样的。故可以先预先把所有的数加起来,若其和和a+b得到的结果相同,那么直接取dp[n][a]+dp[n][b]的值即可,若不同,那么只可以放在任意门中,分别检查一下即可。
代码:
#include <iostream>
#include <cstring>
#include <cstdio>
#define mod 258280327
using namespace std;
int dp[100010][10],arr[100010];
//按给定规则转化为个位数
int modify(int x)
{
int res=0;
while(x)
{
res+=(x%10);
x/=10;
}
if(res>9)
return modify(res);
else
return res;
}
int main()
{
int t,n,a,b,tmp,ans;
int sum,xx;
scanf("%d",&t);
while(t--)
{
sum=0;
//初始化
memset(dp,0,sizeof(dp));
scanf("%d%d%d",&n,&a,&b);
//读入
for(int i=1;i<=n;i++)
scanf("%d",&arr[i]);
for(int i=1;i<=n;i++)
{
//并同时转化为对应的个位数
arr[i]=modify(arr[i]);
sum+=arr[i];
}
sum=modify(sum);
xx=a+b;
xx=modify(xx);
//看是否相等
if(sum==xx)
{
dp[1][arr[1]]=1;
for(int i=2;i<=n;i++)
{
for(int j=1;j<=9;j++)
{
tmp=j-arr[i];
if(tmp<=0)
tmp+=9;
//当前位分别取1-9,由前面对应的关系推过来
dp[i][j]=(dp[i-1][tmp]+dp[i-1][j])%mod;
}
}
printf("%d\n",(dp[n][a]+dp[n][b])%mod);
}
//不相等,只可能单独在一扇门中
else
{
ans=0;
if(sum==a)
ans++;
if(sum==b)
ans++;
printf("%d\n",ans);
}
}
return 0;
}

版权声明:本文为博主原创文章,未经博主允许不得转载。
原文地址:http://blog.csdn.net/david_jett/article/details/47617099