[CF520E] Pluses everywhere

Pluses everywhere

题意：
给你一个$n$位数$a$，让你在这些数之间添加$k$个$+$号，然后得到一个合法表达式，求所有可能的不同的合法表达式之和，答案对$10^9+7$取模。

分析：
题意很简单，但要怎么去想呢？这可以说是一道计数dp的题吧。
我们可以单独的考虑每一位$a_i$的贡献，首先可以肯定的是，一共有$n-1$个位置可以放$+$号，然后我们根据离$a_i$最近且在$a_i$之后的$+$号位置进行讨论：

1. 最近的一个加号在$a_i$后面，也就是说$a_i$作为个位，那么剩下的$k-1$个加号可以有$n-2$个位置填，其贡献为：$a_i\times C_{n-2}^{k-1}$
2. 最近的一个加号在$a_{i+1}$后面，也就是说$a_{i+1}$作为十位，那么剩下的$k-1$个加号可以有$n-3$个位置填，其贡献为：$a_i\times 10\times C_{n-3}^{k-1}$
   …
3. 依次类推，最近的一个加号在$a_{i+n-k-1}$后面，那么剩下的$k-1$个加号可以有$n-2-(n-k-1)==k-1$个位置填，其贡献为：$a_i\times 10^{n-k-1}\times C_{k-1}^{k-1}$
4. 然后假如我们距离$a_i$最近的一个加号在$a_n$后面，因为$a_n$后面不能放加号，所以我们还剩下$k$个加号，$i-1$个位置可以放，其贡献为：$a_i\times 10^{n-i}\times C_{i-1}^k$

那么我们的答案就是这些贡献总和。
化简下可以得到一个最终的公式：${\sum }_{i=1}^{n-k}10^{i-1}\times ({\sum }_{j=1}^{n-i}{a}_j\times C_{n-1-i}^{k-1}+a_{n-i+1}\times C_{n-i}^k)$

代码如下：

#include <bits/stdc++.h>

using namespace std;
typedef long long ll;
const int MOD = 1e9 + 7;
const int maxn = 1e5 + 5;

ll qpow(ll a, ll b, ll mod) {
    ll res = 1;
    while (b) {
        if (b & 1) res = res * a % mod;
        a = a * a % mod;
        b >>= 1;
    }
    return res;
}

ll fac[maxn], inv[maxn];

void init(int n) {
    fac[0] = 1;
    for (int i = 1; i <= n; i++) fac[i] = fac[i - 1] * i % MOD;
    inv[n] = qpow(fac[n], MOD - 2, MOD);
    for (int i = n - 1; i >= 0; i--) inv[i] = inv[i + 1] * (i + 1) % MOD;
}

ll C(int n, int m) {
    if (n < m) return 0;
    return fac[n] * inv[m] % MOD * inv[n - m] % MOD;
}

int a[maxn];
ll dp[maxn], pre[maxn], res, cur = 1ll;

int main() {
    int n, k;
    char ch;
    scanf("%d%d", &n, &k);
    getchar();
    for (int i = 1; i <= n; i++) {
        scanf("%c", &ch);
        a[i] = ch - '0';
        pre[i] = pre[i - 1] + a[i];
    }
    init(n);
    for (int i = 1; i <= n - k; i++) {
        res = (res + cur * (pre[n - i] * C(n - i - 1, k - 1) % MOD +
                    a[n - i + 1] * C(n - i, k) % MOD) % MOD) % MOD;
        cout << pre[n-i]*C(n-i-1,k-1) << " " << a[n-i+1]*C(n-i,k) << endl;
        cur = cur * 10 % MOD;
    }

    printf("%lld\n", res);
    return 0;
}