CF908G New Year and Original Order(DP,数位 DP)

又一次降智……

(数位 DP 原来可以写这么短,学到了)

问题可以转化为求数位中 $\ge k$ 的有恰好 $j$ 位的数的个数。设为 $c_{j,k}$。

那么答案就是:(考虑把 $k$ 的贡献拆开,比如 $9$ 的贡献拆成 $1$ 的贡献的 $9$ 倍,然后分配到 $1$ 到 $9$)

$$\sum_{1\le j\le n,1\le k\le 9}c_{j,k}\underbrace{111\dots111}_{j}$$

求 $c_{j,k}$ 可以数位 DP。(此处不是记忆化搜索的形式,所以方程会有点不同)

$f[i][j][k][l]$ 表示前 $i$ 位,$\ge k$ 的有恰好 $j$ 位,$l$ 是有没有顶到上界(就是后面要不要顶着 $n$ 枚举,类似记忆化搜索中的 $limit$)。

转移较为显然,不再赘述。

时间复杂度 $O(len^2)$。(有一个 $100$ 的常数)

#include<bits/stdc++.h>
using namespace std;
const int maxn=777,mod=1e9+7;
#define FOR(i,a,b) for(int i=(a);i<=(b);i++)
#define ROF(i,a,b) for(int i=(a);i>=(b);i--)
#define MEM(x,v) memset(x,v,sizeof(x))
inline int read(){
    int x=0,f=0;char ch=getchar();
    while(ch<'0' || ch>'9') f|=ch=='-',ch=getchar();
    while(ch>='0' && ch<='9') x=x*10+ch-'0',ch=getchar();
    return f?-x:x;
}
int n,f[maxn][maxn][10][2],ans;
char s[maxn];
int calc(int x){
    int pro=1,s=0;
    FOR(i,1,x){
        s=(s+pro)%mod;
        pro=10ll*pro%mod;
    }
    return s;
}
int main(){
    scanf("%s",s+1);
    n=strlen(s+1);
    FOR(i,0,9) f[0][0][i][0]=1;
    FOR(i,1,n) FOR(j,0,i) FOR(k,0,9) FOR(l,0,9) if(l<k || j){
        if(l==s[i]-'0') f[i][j][k][0]=(f[i][j][k][0]+f[i-1][j-(l>=k)][k][0])%mod;
        f[i][j][k][1]=(f[i][j][k][1]+f[i-1][j-(l>=k)][k][1])%mod;
        if(l<s[i]-'0') f[i][j][k][1]=(f[i][j][k][1]+f[i-1][j-(l>=k)][k][0])%mod;
    }
    FOR(j,0,n) FOR(k,1,9) ans=(ans+1ll*calc(j)*(f[n][j][k][0]+f[n][j][k][1])%mod)%mod;
    printf("%d\n",ans);
}
View Code

 

posted @ 2019-07-19 21:43  ATS_nantf  阅读(174)  评论(0编辑  收藏  举报