题意:
有一位售票员现在要将手中的一系列连续的票卖给客人,而且是卖出连续的票给客人,而现在卖出的票的编号的数位之和不能小于k这个正整数,求按照这样的规则我能售票给多少位乘客。
算法合集之《浅谈数位类统计问题》当中就讲到过,要我们以进制叉树的形式来具象化所有的数的情况。
我们就是要找出l到r中的连续数超过k的组数,尽可能的多。
但是现在关于合并子树的问题就是,其中有一部分客人手机的票超过了k,并找有一部分超过k的部分,那么这就使得我们再累加子树的时候添加了困难的地方。
现在引入一个dp的思想,说实话此类的问题这样的思想还是比较先进的,就是难得想到。
Dp[i][j][m]其中有co,rest两个值,其中表示后i数字任意,前面累加起来的和是j,区间前一部分剩余的和是m,co表示该区间内部分组数量,rest表示区间尾部剩余的和。
代码:
View Code
1 #include<iostream> 2 #include<cstring> 3 #include<cstdlib> 4 #include<cstdio> 5 #include<cmath> 6 using namespace std; 7 typedef long long LL; 8 int const N = 20; 9 struct node 10 { 11 LL co; 12 int rest; 13 }dp[N][200][1000]; 14 LL l,r,ans; 15 int n,a[N],b[N],sa[N],sb[N],ta,tb,k,rest; 16 void create_dp() 17 { 18 for(int i=0;i<=9*n;i++) 19 for(int j=0;j<k;j++) 20 if(i+j<k)dp[0][i][j].co=0,dp[0][i][j].rest=i+j; 21 else dp[0][i][j].co=1,dp[0][i][j].rest=0; 22 for(int i=1;i<=n;i++) 23 { 24 for(int j=0;j<=9*n;j++) 25 { 26 for(int m=0;m<k;m++) 27 { 28 rest=m; 29 for(int c=0;c<=9;c++) 30 { 31 dp[i][j][m].co+=dp[i-1][j+c][rest].co; 32 rest=dp[i-1][j+c][rest].rest; 33 } 34 dp[i][j][m].rest=rest; 35 } 36 } 37 } 38 } 39 void getsum_r(int t) 40 { 41 int s=a[1]; 42 for(int i=1;i<=t;i++) 43 { 44 for(int j=s;j<=9;j++) 45 { 46 ans+=dp[i-1][sa[i+1]+j][rest].co; 47 rest=dp[i-1][sa[i+1]+j][rest].rest; 48 } 49 s=a[i+1]+1; 50 } 51 } 52 void getsum_l(int t,int y) 53 { 54 for(int i=t;i>0;i--) 55 { 56 int down=(i==y?1:0); 57 int up=(i==1?b[i]:b[i]-1); 58 for(int j=down;j<=up;j++) 59 { 60 ans+=dp[i-1][sb[i+1]+j][rest].co; 61 rest=dp[i-1][sb[i+1]+j][rest].rest; 62 } 63 } 64 } 65 void solution() 66 { 67 ta=tb=0; 68 rest=0; 69 LL num=l; 70 for(;num;num/=10)a[++ta]=num%10; 71 num=r; 72 for(;num;num/=10)b[++tb]=num%10; 73 sa[ta+1]=0; 74 for(int i=ta;i>0;i--)sa[i]=sa[i+1]+a[i]; 75 sb[tb+1]=0; 76 for(int i=tb;i>0;i--)sb[i]=sb[i+1]+b[i]; 77 if(ta<tb) 78 { 79 getsum_r(ta); 80 for(int i=ta+1;i<tb;i++) 81 { 82 for(int j=1;j<=9;j++) 83 { 84 ans+=dp[i-1][j][rest].co; 85 rest=dp[i-1][j][rest].rest; 86 } 87 } 88 getsum_l(tb,tb); 89 } 90 else 91 { 92 while(ta>1&&a[ta]==b[ta])ta--; 93 getsum_r(ta-1); 94 int down=(ta==1?a[ta]:a[ta]+1); 95 int up=(ta==1?b[ta]:b[ta]-1); 96 for(int i=down;i<=up;i++) 97 { 98 ans+=dp[ta-1][sa[ta+1]+i][rest].co; 99 rest=dp[ta-1][sa[ta+1]+i][rest].rest; 100 } 101 getsum_l(ta-1,tb); 102 } 103 } 104 int main() 105 { 106 cin>>l>>r>>k; 107 LL num=r; 108 n=0; 109 while(num>0)num/=10,n++; 110 ans=0; 111 create_dp(); 112 solution(); 113 cout<<ans<<endl; 114 return 0; 115 }