CodeForces 520 E.Pluses everywhere（组合数学+dp）

Description

给出一个长度为$n$的数字串，现在要往其中加$k$个$+$将其变成表达式，要求$+$把串分成$k$个数字，可以有前导零，每个数字长度至少为$1$，对于一个表达式可以计算其值，问所有方案对应的值的和

Input

第一行输入两个整数$n,k$表示串的长度和要加入的$+$数量，之后输入一个长度为$n$的数字串$(1\le k<n\le 10^5)$

Output

输出所有方案对应的值的和，结果模$10^9+7$

Sample Input

3 1
108

Sample Output

27

Solution

设$n$个数字从高位到低位为$a_0,...,a_{n-1}$，考虑$a_i$对答案的影响时不用考虑$a_i$前面加号的位置，只需要考虑后面第一个加号的位置，设后面第一个加号的位置在$a_j$和$a_{j+1}$之间，则其对答案的贡献为$10^{j-i}$，而还有$n+i-j-2$个位置要放$k-1$个加号，方案数$C_{n+i-j-2}^{k-1}$，还有一种情况是$a_i$后面没有加号，其对答案的贡献是$10^{n-i-1}$，那么前面$i$个位置要放$k$个加号，方案数$C_i^k$，故$ans=\sum\limits_{i=0}^{n-1}a_i\sum\limits_{j=i}^{n-2}10^{j-i}\cdot C_{n+i-j-2}^{k-1}+\sum\limits_{i=0}^{n-1}a_i\cdot 10^{n-i-1}\cdot C_i^k$，预处理$10^x$和组合数，后一部分可以$O(n)$得到，前一部分要化简一下，考虑枚举$j-i$则前一部分可以变成$\sum\limits_{i=0}^{n-2}10^i\cdot C_{n-i-2}^{k-1}\sum\limits_{j=0}^{n-2-i}a_j$，令$s_i=\sum\limits_{j=0}^{i}a_j$，则可以$O(n)$得到第一部分答案

Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
typedef pair<int,int>P;
const int INF=0x3f3f3f3f,maxn=100005;
#define mod 1000000007
int fact[maxn],inv[maxn],f[maxn];
void init(int n=1e5)
{
    fact[0]=1;
    for(int i=1;i<=n;i++)fact[i]=(ll)i*fact[i-1]%mod;
    inv[1]=1;
    for(int i=2;i<=n;i++)inv[i]=mod-(ll)(mod/i)*inv[mod%i]%mod;
    inv[0]=1;
    for(int i=1;i<=n;i++)inv[i]=(ll)inv[i-1]*inv[i]%mod;
    f[0]=1;
    for(int i=1;i<=n;i++)f[i]=(ll)10*f[i-1]%mod;
}
int n,k,a[maxn],s[maxn];
char c[maxn];
int C(int n,int m)
{
    if(n<m||m<0)return 0;
    return (ll)fact[n]*inv[m]%mod*inv[n-m]%mod;
}
void add(int &x,int y)
{
    x=x+y>=mod?x+y-mod:x+y;
}
int main()
{
    init();
    while(~scanf("%d%d",&n,&k))
    {
        scanf("%s",c);
        for(int i=0;i<n;i++)a[i]=c[i]-'0',s[i]=(i?s[i-1]:0)+a[i];
        int ans=0;
        for(int i=0;i<=n-2;i++)add(ans,(ll)s[n-2-i]*f[i]%mod*C(n-2-i,k-1)%mod);
        for(int i=0;i<n;i++)add(ans,(ll)a[i]*f[n-1-i]%mod*C(i,k)%mod);
        printf("%d\n",ans);
    }   
    return 0;
}