码迷,mamicode.com
首页 > 其他好文 > 详细

20140712 classic 数位DP

时间:2014-07-13 23:36:05      阅读:269      评论:0      收藏:0      [点我收藏+]

标签:style   blog   http   color   os   数据   

其实就是一道蛮简单的数位DP

考试的时候出了点小错导致基本Wa0 

还好数据分治有30分- -

num[i][j][k]表示前i位数字和为j的数的个数  k=0表示不顶上界  k=1表示顶上界

转移方程见代码

dp[i][j][k]表示前i位数字和为j的数的和

转移方程同见代码

bubuko.com,布布扣
 1 #include <cstdio>
 2 #include <cstring>
 3 using namespace std;
 4 #define mod 1000000007
 5 typedef long long ll;
 6 
 7 ll dp[60][600][2];
 8 ll num[60][600][2];
 9 int x,y;
10 
11 ll DP(char ch[60]) {
12     memset(dp,0,sizeof(dp));
13     memset(num,0,sizeof(num));
14     int len=strlen(ch);
15     int a[60];
16     int sum[60];
17     ll ans=0;
18     sum[0]=0;
19     for (int i=0;i<len;i++) {
20         a[i+1]=(int)(ch[i])-48;
21         sum[i+1]=sum[i]+a[i+1];
22     }
23     for (int j=0;j<a[1];j++) {
24         dp[1][j][0]=j; num[1][j][0]=1;
25     }
26     dp[1][a[1]][1]=a[1]; num[1][a[1]][1]=1;
27     for (int i=1;i<len;i++) {
28         for (int j=0;j<600;j++) {
29             if (j!=0 && dp[i][j][0]==0) continue;
30             for (int k=0;k<10;k++) {
31                 num[i+1][j+k][0]=(num[i+1][j+k][0]+num[i][j][0])%mod;
32                 dp[i+1][j+k][0]=(dp[i+1][j+k][0]+dp[i][j][0]*10+k*num[i][j][0])%mod;
33             }
34         }
35         if (sum[i]!=0 && dp[i][sum[i]][1]==0) continue;
36         for (int k=0;k<a[i+1];k++) {
37             num[i+1][sum[i]+k][0]=(num[i+1][sum[i]+k][0]+num[i][sum[i]][1])%mod;
38             dp[i+1][sum[i]+k][0]=(dp[i+1][sum[i]+k][0]+dp[i][sum[i]][1]*10+k*num[i][sum[i]][1])%mod;
39         }
40         num[i+1][sum[i+1]][1]=(num[i][sum[i]][1]+num[i+1][sum[i+1]][1])%mod;
41         dp[i+1][sum[i+1]][1]=(dp[i+1][sum[i+1]][1]+dp[i][sum[i]][1]*10+a[i+1]*num[i][sum[i]][1])%mod;
42     }
43     for (int j=x;j<=y;j++) ans=(ans+dp[len][j][0]+dp[len][j][1])%mod;
44     return ans;
45 }
46 
47 char l[60],r[60];
48 
49 int main() {
50     scanf("%s",&l);
51     scanf("%s",&r);
52     scanf("%d%d",&x,&y);
53     int len=strlen(l);
54     len--;
55     while (1) {
56         if (l[len]==0) l[len]=9;
57         else {
58             l[len]=(char(int(l[len])-1));
59             break;
60         }
61         len--;
62         if (len==0) {
63             for (int i=strlen(l)-1;i>0;i--) {
64                 l[i-1]=l[i];
65             }
66             l[strlen(l)-1]=\n;
67             break;
68         }
69     }
70     ll R=DP(r);
71     ll L=DP(l);
72     while (R-L<0) R+=mod;
73     printf("%lld",(R-L)%mod);
74 }
View Code

 

20140712 classic 数位DP,布布扣,bubuko.com

20140712 classic 数位DP

标签:style   blog   http   color   os   数据   

原文地址:http://www.cnblogs.com/fjmmm/p/3840398.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!