可（数位dp）

dygxczn

已于 2023-11-16 15:52:13 修改

阅读量101

点赞数 4

文章标签：算法

于 2023-10-13 09:12:50 首次发布

本文链接：https://blog.csdn.net/dygxczn/article/details/133779328

版权

题目

设 $g (n)$ 为 $\sum\limits_{i=1}^ka_i=n$ 的方案数，答案即为 $\sum\limits_{i=0}^{kx}f(i)g(i)$

考虑容斥去求 $g (n)$ 。
$g(n)=\sum\limits_{i=0}^k(-1)^i\binom{k}{i}\binom{n-i(x+1)+k-1}{k-1}$

简单来说就是枚举有 $\ge i$ 个数是大于 $x$ 的，在 $k$ 个数中选，然后把这 $k$ 个数都减去 $x + 1$ （减完后剩余的部分任意选），用插板法求出方案数（ $\sum\limits_{i=1}^nx_i=m$ 非负整数解的数量是 $\binom{n+m-1}{n-1}$ ）。

代入答案式子
$\begin{aligned} \sum\limits_{i=0}^{kx}f(i)g(i)&=\sum\limits_{i=0}^{kx}f(i)\sum\limits_{j=0}^k(-1)^j\binom{k}{i}\binom{n-j(x+1)+k-1}{k-1}\\ &=\sum\limits_{j=0}^k(-1)^j\binom{k}{i}\sum\limits_{i=0}^{kx}f(i)\binom{n-j(x+1)+k-1}{k-1}\\ &=\dfrac{1}{(k-1)!}\sum\limits_{j=0}^k(-1)^j\binom{k}{i}\sum\limits_{l=0}^{k-1}h_{j,l}\sum\limits_{i=0}^{kx}f(i)i^l\\ \end{aligned}$

其中 $h_{j,l}=[i^l](i-(x+1)j+k-1)^{k-1\over}$ 。把 $i$ 看作多项式中的变量，那么 $h_{j,l}$ 就是 $l$ 次的系数。

若 $j$ 是常量，我们可以用 $D P$ 在 $O(k^2)$ 的时间求出所有 $h_{j,l}$ 。就考虑前 $i$ 个数，选了 $j$ 个的全部方案的和。

这里有个问题，如果 $i < T$ ，组合数拆成下降幂就会有问题。所以 $i$ 有下界 $T = j (x + 1)$ 。

问题转换为求 $\sum\limits_{i=T}^{kx}f(i)i^l$ 。

在这里，为了让求值更方便，考虑 $i > k x$ 时 $g (i)$ 是多少，显然由定义和容斥得 $g (i) = 0$ 。所以上界可以忽略，为了下面数位 $D P$ 方便，将上界设为 $999999\dots$ （大于等于 $k x$ ）。

先考虑没有下界的情况。

设 $dp_{j}=\sum f(i)i^j$ ，从高位往低位 $D P$ ， $n o w$ 是当前位， $a$ 是当前位填什么数字。

$\begin{aligned} dp_{j}&=\sum\limits_{a=0}^9\sum f(i)f(a)(i+a\times10^{now})^j\\ &=\sum\limits_{a=0}^9\sum f(i)f(a)\sum\limits_{l=0}^j\binom{j}{l}i^{j-l}(a\times10^{now})^l\\ &=\sum\limits_{a=0}^9\sum\limits_{l=0}^j\binom{j}{l}f(a)(a\times10^{now})^{j-l}dp_{l} \end{aligned}$

时间复杂度 $O(10k^2\lg x)$ 。

对于有下界的情况，设 $f_{o,l}=\sum\limits_{i=T}f(i)i^l$ （前 $o$ 位）， $g_{o,l}=\sum\limits_{i>T}f(i)i^l$ （前 $o$ 位），同样从高位往低位转移。

设 $T$ 当前位的数字为 $A$ ， $c_{i,j}=f(i)(i\times10^{now})^j$ ， $d_{i,j}=\sum\limits_{l=i}^{k-1}c_{l,j}$ （后缀和）。

先给出转移
$f_{i,j}=\sum\limits_{l=0}^j\binom{j}{l}c_{A,j-l}f_{i+1,l}$
$g_{i,j}=\sum\limits_{l=0}^j\binom{j}{l}d_{0,j-l}\cdot g_{i+1,l}+\sum\limits_{l=0}^j\binom{j}{l}d_{A+1,j-l}\cdot f_{i+1,l}$

$f_{i,j}$ 的转移和上面 $d p$ 是一样的，只不过不用枚举当前位填什么数字，就填 $A$ 。

$g_{i,j}$ 由于 $i$ 有下界 $T$ ，若前面的位已经比 $T$ 大，后面无论填什么数字都比 $T$ 大，所以加上 $\binom{j}{l}d_{0,j-l}\cdot g_{i+1,l}$ ；如果前面的位和 $T$ 的一样，那么这一位填比 $A$ 大的数，也是比 $T$ 大的，所以加上 $\binom{j}{l}d_{A+1,j-l}\cdot f_{i+1,l}$ 。

至此，我们已经解决这个问题，总的时间复杂度为 $O(k^3\lg n)$

具体实现参见代码

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll mod=1e9+7;
const int N=1e4+10,K=25;
int n,k;
ll f[N][K],g[N][K],C[K][K],ans,b[N],pw[N],a[N],h[K],c[K][K],d[K][K];
char s[N];
ll ksm(ll a,ll b)
{
    ll ans=1;
    while(b){
        if(b&1) ans=ans*a%mod;
        b>>=1;
        a=a*a%mod;
    }
    return ans;
}
void init()
{
    C[0][0]=1;
    for(int i=1;i<=k;i++){
        C[i][0]=1;
        for(int j=1;j<=i;j++){
            C[i][j]=(C[i-1][j]+C[i-1][j-1])%mod;
        }
    }
    for(int i=0;i<n;i++) b[i]=s[n-i-1]^48;
    n+=3;
    b[0]++;
    for(int i=0;i<=n;i++) b[i+1]+=b[i]/10,b[i]%=10;
    pw[0]=1;
    for(int i=1;i<=n;i++) pw[i]=pw[i-1]*10%mod;
}
ll solve()
{
    memset(f,0,sizeof(f));
    memset(g,0,sizeof(g));
    f[n+1][0]=1;
    for(int K=n,A;K>=0;K--){
        A=a[K];
        for(int i=9,X,Y;i>=0;i--){
            X=max(i,1),Y=i*pw[K]%mod;
            c[i][0]=d[i][0]=X;
            d[i][0]=(d[i][0]+d[i+1][0])%mod;
            for(int j=1;j<k;j++){
                c[i][j]=d[i][j]=c[i][j-1]*Y%mod;
                d[i][j]=(d[i][j]+d[i+1][j])%mod;
            }
        }
        for(int i=k-1;i>=0;i--){
            for(int j=0;j<=i;j++){
                (g[K][i]+=g[K+1][j]*C[i][j]%mod*d[0][i-j])%=mod;
                (g[K][i]+=f[K+1][j]*C[i][j]%mod*d[A+1][i-j])%=mod;
            }
            for(int j=0;j<=i;j++){
                (f[K][i]+=f[K+1][j]*C[i][j]%mod*c[A][i-j])%=mod;
            }
        }
    }
    ll num=0;
    for(int i=n;i>=0;i--) num=(num+pw[i]*a[i])%mod;
    memset(h,0,sizeof(h));
    h[0]=1;
    for(int i=1;i<k;i++){
        for(int j=i;j>=1;j--){
            h[j]=(h[j]+h[j-1]*(i-num+mod))%mod;
        }
    }
    ll ans=0;
    for(int i=0;i<k;i++) ans=(ans+h[k-1-i]*(f[0][i]+g[0][i]))%mod;
    return ans;
}
int main()
{
    scanf("%d%s",&k,s);
    n=strlen(s);
    init();
    for(int i=0;i<=k;i++){
        ans=(ans+(i&1?-1:1)*C[k][i]*solve())%mod;
        for(int j=0;j<=n;j++){
            a[j]+=b[j];
            a[j+1]+=a[j]/10,a[j]%=10;
        }
    }
    for(int i=1;i<k;i++) ans=ans*ksm(i,mod-2)%mod;
    cout<<(ans+mod)%mod;
}