定义状态f[i][j][k][x]表示有多少个i位数t有j位大于x,k位等于x,然而复杂度无法接受,发现可以改变状态为f[i][j][x]表示前i为有j个数字大于等于x的方案数,就可以快速转移了
1 #include<bits/stdc++.h> 2 using namespace std; 3 #define mod 1000000007 4 int n,ans,f[1005][1005][11][2]; 5 char s[1005]; 6 int main(){ 7 scanf("%s",s); 8 n=strlen(s); 9 for(int i=0;i<10;i++)f[0][0][i][0]=1; 10 for(int i=0;i<n;i++){ 11 int p=(s[i]-'0'); 12 for(int j=0;j<=i;j++) 13 for(int k=0;k<10;k++) 14 for(int l=0;l<2;l++) 15 for(int t=0;t<=max(p,9*l);t++){ 16 f[i+1][j+(k<=t)][k][l|(t<p)]+=f[i][j][k][l]; 17 f[i+1][j+(k<=t)][k][l|(t<p)]%=mod; 18 } 19 } 20 for(int i=1,s=0;i<=n;i++){ 21 s=(10LL*s+1)%mod; 22 for(int j=1;j<10;j++) 23 for(int k=0;k<2;k++)ans=(ans+1LL*s*f[n][i][j][k])%mod; 24 } 25 printf("%d",ans); 26 }