【CF908G】 New Year and Original Order

题目

题目描述
Let S(n)S(n) denote the number that represents the digits of nn in sorted order. For example, S(1)=1,S(5)=5,S(50394)=3459,S(353535)=333555S(1)=1,S(5)=5,S(50394)=3459,S(353535)=333555 .

Given a number XX , compute modulo 10^{9}+710
9
+7 .

输入格式
The first line of input will contain the integer XX ( 1<=X<=10^{700}1<=X<=10
700
).

输出格式
Print a single integer, the answer to the question.

题意翻译
给n<=10^{700}n<=10
700
，问1到n中每个数在各数位排序后得到的数的和。答案膜1e9+7。 Translated by @远航之曲

输入输出样例
输入 #1复制
21
输出 #1复制
195
输入 #2复制
345342
输出 #2复制
390548434
说明/提示
The first few values of SS are 1,2,3,4,5,6,7,8,9,1,11,12,13,14,15,16,17,18,19,2,121,2,3,4,5,6,7,8,9,1,11,12,13,14,15,16,17,18,19,2,12 . The sum of these values is 195195 .

思路

设f[i][j][k][0/1]表示前i位中有j位的数字大于等于k，是否小于n的方案数。

然后枚举第i位的数字p，直接大力转移。

考虑一下怎么计算答案。

实际上前nn位有ii位数字大于等于kk的方案数s=f[n][i][k][0]+f[n][i][k][1]

每一个s对ans的贡献是sum*( 满足 k≤i 的数字的个数 ).

​

代码

#include<bits/stdc++.h>
using namespace std;
const int mod=1e9+7,N=777;
typedef long long ll;
ll f[N][N][2];
char s[N];
ll ans=0;
int main() {
    scanf("%s",s+1); int len=strlen(s+1);
    for(int k=1;k<=9;k++){
        memset(f,0,sizeof(f));
        for(int i=0; i<=s[1]-'0'; i++) 
		{
            bool ty2=(i>=k),ty3=(i==(s[1]-'0'));
            f[1][ty2][ty3]+=1;
        }
        for(int i=1; i<len; i++) {
            for(int j=0; j<len; j++) {
                for(int st=0;st<=1;st++) {
                    if(!f[i][j][st])continue;
                    for(int add=0;add<=(st?(s[i+1]-'0'):9);add++) {
                        f[i+1][j+(add>=k)][st&&(add==(s[i+1]-'0'))]+=f[i][j][st];
                        f[i+1][j+(add>=k)][st&&(add==(s[i+1]-'0'))]%=mod;
                    }
                }
            }
        }
        ll tmp=0;
        for(int j=0; j<=len; j++) {
            ans+=(f[len][j][0]+f[len][j][1])%mod*tmp%mod;
            ans%=mod;tmp=(tmp*10)+1;tmp%=mod;
        }

    }
    printf("%lld",ans);
    return 0;
}