2021 杭电多校赛第8场 1005 Separated Number

题目链接

题意

给定一个位数为$n(1\le n\le 10^6)$的正整数
每一次操作你可以将该整数切成不超过$k(1\le k\le n)$各部分
计算所有切片方案的生成的数的和
结果对$998244353$取模

思路

这个答案直接计算比较困难，我们考虑计算每一位在切片完成后做是个位、十位…的方案数。因此我们就有了某一位在结果位第$i$位的方案数
$$\sum _{j=0}^{k-2}{C}_{n-1-i}^j$$
而且我们惊喜的发现根据这个公式
$$C_{n-1}^{m-1}+C_{n-1}^m=C_n^m$$
可以地推算出
$$\sum _{j=0}^{k-2}{C}_{n-1-i}^j=\frac{{\sum }_{j=0}^{k-2}{C}_{n-i}^j+C_{n-i}^{k-2}}{2}$$
这就可以在$i=1$的时候暴力计算，然后递推计算$i>1$的结果了。
到这里这道题的大部分其实都做完了。
但还有一个一个小地方需要特殊处理
对于最后一位来说，他是永远只能做个位的，方案数应该为
$$\sum _{j=0}^{k-2}{C}_{n-1}^j$$
同理递推计算倒数第$i$位做第$i$位的方案数。

参考代码

#include <bits/stdc++.h>

using namespace std;
const int maxn=1e6+10;
const int mod=998244353;
typedef long long ll;
char s[maxn];
int a[maxn];

ll sum[maxn],pw[maxn],fac[maxn],inv[maxn];
ll cnt[maxn],inv2;
ll kpow(ll a,ll k){
    ll res=1;
    while(k){
        if(k&1) res=res*a%mod;
        a=a*a%mod;
        k>>=1;
    }
    return res;
}
ll cal(ll n,ll m){
    if(n<m) return 0;
    return fac[n]*inv[n-m]%mod*inv[m]%mod;
}
void solve(){
    int k;scanf("%d",&k);
    scanf("%s",s+1);
    int n=strlen(s+1);
    for(int i=0;i<=n+1;i++) cnt[i]=0;
    for(int i=1;i<=n;i++){
        a[i]=s[i]-'0';
    }
    for(int i=1;i<=n;i++){
        sum[i]=sum[i-1]+a[i];
    }
    ll ans=0;
    for(int i=1;i<n;i++){
        if(i==1){
            for(int j=0;j<=min(k-2,n-1-i);j++) cnt[i]=(cnt[i]+cal(n-1-i,j))%mod;
        }
        else {
            cnt[i]=(cnt[i-1]+cal(n-1-i,k-2))%mod*inv2%mod;
        }
        ans=(ans+cnt[i]*pw[i-1]%mod*sum[n-i]%mod)%mod;
    }
    ll last=0;
    for(int i=n;i>=1;i--){
        if(i==n) {
            for(int j=0;j<=min(k-1,n-1);j++) last=(last+cal(n-1,j))%mod;
        }
        else {
            last=(last+cal(i-1,k-1))%mod*inv2%mod;
        }
        ans=(ans+last*pw[n-i]%mod*a[i]%mod)%mod;
    }
    printf("%lld\n",ans);
}
void init(){
    pw[0]=1;
    fac[0]=1;
    inv[0]=1;
    inv2=kpow(2,mod-2);
    for(int i=1;i<maxn;i++){
        pw[i]=pw[i-1]*10%mod;
        fac[i]=fac[i-1]*i%mod;
        inv[i]=kpow(fac[i],mod-2);
    }
    
}
int main(){
//    freopen("in.txt","r",stdin);
    init();
    int T;scanf("%d",&T);while(T--)
    solve();
    return 0;
}