[cf908G]New Year and Original Order

定义状态f[i][j][k][x]表示有多少个i位数t有j位大于x,k位等于x,然而复杂度无法接受,发现可以改变状态为f[i][j][x]表示前i为有j个数字大于等于x的方案数,就可以快速转移了

[cf908G]New Year and Original Order
 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 }
View Code

 

上一篇:[POJ-3613]Cow Relays


下一篇:LeetCode 1005 Longest Palindromic Substring