杭电多校第八场05_Separated Number（组合数前缀和性质）

题目

Separated Number

给出一个数位为$n$的数字$(n<=1e6)$，现在可以将该数字最多分成$k(1<=k<=n)$段。定义一种分法的贡献为所有段的数字之和，求所有分法的贡献和。

例如对于样例，答案为$1+1+10+100=112$

3
100

1|0|0 :1+0+0=1  
1|00  :1+0=1  
10|0  :10+0=10  
100   :100

比较容易想到的一种解法是对于每一位，计算它作为个位、十位… …的贡献和。

例如要对于数字$123456$，最多可以分成$6$段，要计算$3$作为十位的贡献次数。 我们需要在$4$后面分割一次，还剩下最多$k-2$次分割机会和$n-3$个可分割点，方案数显然就是$C_{n-3}^0+C_{n-3}^1+......+C_{n-3}^{k-2}$。

直接计算方案数时间复杂度过高，需要先掌握组合数前缀和的知识。

递推组合数前缀和

我们先来证明以下公式

$C_n^m=C_{n-1}^{m-1}+C_{n-1}^m$

证明：

$C_n^m=\frac{(n)!}{(n-m)!\times (m)!}$

$C_{n-1}^{m-1}+C_{n-1}^m=\frac{(n-1)!}{(n-m)!\times (m-1)!}+\frac{(n-1)!}{(m)!\times (n-m-1)!}=\frac{m\times (n-1)!}{(n-m)!\times (m)!}+\frac{(n-m)\times (n-1)!}{(n-m)!\times (m)!}=\frac{(n)!}{(n-m)!\times (m)!}=C_n^m$

上面的公式在杨辉三角递推组合数中有所应用。

记$F(n,m)={\sum }_{i=0}^m{C}_n^m$，可以发现，这在杨辉三角中体现为某一列的前缀和。

有以下两个递推式：

$$F(n,m)=F(n,m-1)+C(n,m)$$

$$F(n,m)=2\times F(n-1,m)-C(n-1,m)$$

第一个公式很显然，前缀和的性质，也就是杨辉三角某一行前$m-1$个数的和再加上第$m$个数，显然就是前$m$个数的和。

第二个公式也不难，回想一下杨辉三角递推组合数的方式，除了最后一个数，其他数都被加了两遍。

好了， 到这为止，我们已经掌握了递推组合数前缀和。

解题

考虑算每一位对答案产生的贡献，假设这一位为$d(0<=d<=9)$，那么他能产生的贡献就是$d\times 10^j\times num$，其中，$num$为让$d$产生$10^j$的位权的分法总方案数。

要使$d$产生$10^j$的位权，记$d$的下标为$i$，那么我们应在第$i+j$位后面进行一次分割，剩下的分割点和分割次数就会产生若干方案数。

将$i+j$分三种情况讨论：

1. $i+j>n$，不合法的方案，无贡献。
2. $i+j=n$，合法的方案，$i+j$这个分割点是“本来就有的”，不花费分割次数，因此方案数为${\sum }_{i=0}^{k-1}{C}_{n-j-1}^i$（$n-j-1$个分割点，$k-1$次分割机会）
3. $i+j<n$，合法的方案，需要在位$i+j$后面进行一次分割，因此方案数为${\sum }_{i=0}^{k-2}{C}_{n-j-2}^i$

对于情况2，若几个不同的位，位权$10^j$都存在，那么这几个不同的位以情况2的方式产生$10^j$的位权的方案数是一样的。并且如果位置$i$存在产生位权$10^j$的方案，那么$i$前面的位置也一定存在，这样一来，我们从后往前考虑每一位，对情况2的方案数做后缀和，每一位只需计算新增的贡献，优化了时间复杂度。

对于情况3，我们直接计算即可。

我们需要用的是$F(i,k-1)$和$F(i,k-2)$，$i$从$0$到$n-1$，可以通过上面的组合数前缀和性质进行递推。

代码

#include <bits/stdc++.h>
using namespace std;

const long long mod = 998244353;
const int N = 1e6 + 6;
char s[N];

long long qpow(long long a, long long b)
{
    long long res = 1;
    while (b)
    {
        if (b & 1)
            res = res * a % mod;
        b /= 2;
        a = a * a % mod;
    }
    return res;
}
long long fact[N];
void init()
{
    fact[0] = 1;
    for (int i = 1; i < N; i++)
        fact[i] = fact[i - 1] * i % mod;
}
long long inv(long long x)
{
    return qpow(x, mod - 2);
}
long long C(long long n, long long m)
{
    if (n < m)
        return 0;
    return fact[n] * inv(fact[m]) % mod * inv(fact[n - m]) % mod;
}

long long pwd10[N];
long long A[N]; //预处理 F(n,k-2)
long long B[N]; //预处理 F(n,k-1)

long long n, k;
void solve()
{

    pwd10[0] = 1;
    for (int i = 1; i < n; i++)
        pwd10[i] = pwd10[i - 1] * 10 % mod;
    A[0] = k - 2 >= 0 ? 1 : 0;
    B[0] = k - 1 >= 0 ? 1 : 0;
    for (int i = 1; i < n; i++)
    {
        A[i] = (2 * A[i - 1] - C(i - 1, k - 2) + mod) % mod;
        B[i] = (2 * B[i - 1] - C(i - 1, k - 1) + mod) % mod;
    }

    long long ans = 0, sum = 0;
    for (int i = n; i >= 1; i--)
    {
        long long val = s[i] - '0';
        if (i != n)
            sum = (sum + A[i - 1] * pwd10[n - i - 1]) % mod;
        ans = (ans + val * (B[i - 1] * pwd10[n - i] % mod + sum)) % mod;
    }
    cout << ans << endl;
}
int main()
{
    init();
    int t;
    scanf("%d", &t);
    while (t--)
    {
        scanf("%lld %s", &k, s + 1);
        n = strlen(s + 1);
        solve();
    }
    return 0;
}