CodeForces-520E Pluses everywhere

题目描述

给出一个长度为 $n$ 的字符串，给出一个非负整数 $k$，要求给这个字符串中间添加 $k$ 个$$+$’号，变成一个表

达式，比如”$1000101$”,添加两个$＋$号，可以变成”$10+001+01$”,或者”$1000+1+01$”,表达式的值分别是$12$ 和 $1002$。

问所有的添加加号的方案的表达式的值的和是多少。

Input

两个整数 $n，k$，一个字符串$ s$ $(0<=k<n<=1e5).$

Output

一个整数，模$ 1000000007$

Sample Input

Sample Output

27

9

计数$DP$

首先，看到那么大的数据范围肯定是对于每个数计算贡献。

那么我们该如何计算贡献呢？

对于题目给出的那个字符串$S$，从左往右每个数字依次为$a_{n-1},a_{n-2}....a_{1},a_0$.

现在，我们对于$a_t$，讨论其贡献。

$a_t$右边有$t$个位置可以放置加号，同时一共有$n-1$个位置放置加号。

若$a_t$右边第一个位置放置了加号，则$a_t$被当成个位，还有$n-2$个空位，$k-1$个加号，一共有$C(k-1,n-2)$种方案。

其贡献为$C(k-1,n-1)*10^{0}*a_t$。

若$a_t$右边第一个位置为空，第二个位置放置加号，则$a_t$为十位，还有$n-3$个空位，$k-1$个加号，共有$C(k-1,n-3)$种方案。

贡献为$C(k-1,n-2)*10^{1}*a_t$.

依次类推，

若$a_t$右边为空，则贡献为$C(k,n-t-2)*10^{t}*a_t$。

所以，$a_t$这个数的贡献为$(10^{t}*C(k,n-t-2)+\sum^{t}_{j=0}10^{j}*C(k-1,n-j-2))*a_t$。

于是我们可以得到第$i$个数的贡献计算公式$A_i=10^{i}*C(k,n-i-2)+\sum^{i}_{j=0}10^{j}*C(k-1,n-j-2)$。

最后，$Ans=\sum^{n-1}_{i=0}A_i*a_i$。

代码如下

#include <cstdio>

#include <cstring>

#include <iostream>

#include <algorithm>

using namespace std;

#define int long long

#define reg register

#define Raed Read

#define clr(a,b) memset(a,b,sizeof a)

#define Mod(x) (x>=mod)&&(x-=mod)

#define debug(x) cerr<<#x<<" = "<<x<<endl;

#define max(a,b) ((a)>(b)?(a):(b))

#define min(a,b) ((a)>(b)?(b):(a))

#define rep(a,b,c) for(reg int a=(b),a##_end_=(c); a<=a##_end_; ++a)

#define ret(a,b,c) for(reg int a=(b),a##_end_=(c); a<a##_end_; ++a)

#define drep(a,b,c) for(reg int a=(b),a##_end_=(c); a>=a##_end_; --a)

#define erep(i,G,x) for(int i=(G).Head[x]; i; i=(G).Nxt[i])

#pragma GCC target("avx,avx2,sse4.2")

#pragma GCC optimize(3)

inline int Read(void) {

	int res=0,f=1;

	char c;

	while(c=getchar(),c<48||c>57)if(c=='-')f=0;

	do res=(res<<3)+(res<<1)+(c^48);

	while(c=getchar(),c>=48&&c<=57);

	return f?res:-res;

}

template<class T>inline bool Min(T &a, T const&b) {

	return a>b?a=b,1:0;

}

template<class T>inline bool Max(T &a, T const&b) {

	return a<b?a=b,1:0;

}

const int N=1e5+5,M=1e6+5,mod=1e9+7;

bool MOP1;

int Fac[N],Inv[N];

inline int Pow(int x) {

	int res=1,y=mod-2;

	while(y) {

		if(y&1)res=(res*x)%mod;

		x=(x*x)%mod,y>>=1;

	}

	return res;

}

inline int C(int x,int y) {

	if(!x)return 1;

	return (Fac[y]*((Inv[x]*Inv[y-x])%mod))%mod;

}

int A[N],Pow_10[N];

char S[N];

bool MOP2;

inline void _main() {

	int n=Read(),k=Read(),Ans=0,tot=0;

	scanf("%s",S),Fac[0]=Pow_10[0]=Inv[0]=1;

	if(!k) {

		int Ans=0;

		ret(i,0,n)Ans=(Ans*10+(S[i]^48))%mod;

		printf("%lld\n",Ans);

		return;

	}

	rep(i,1,n) {

		Fac[i]=(Fac[i-1]*i)%mod;

		Pow_10[i]=(Pow_10[i-1]*10)%mod;

		Inv[i]=Pow(Fac[i]);

	}

	A[0]=C(k-1,n-2);

	int P=n-k-1;

	rep(i,0,P)A[i]=(A[i-1]+Pow_10[i]*C(k-1,n-i-2))%mod;

	rep(i,P+1,n)A[i]=A[i-1];

	rep(i,0,P)A[i]=(A[i]+Pow_10[i]*C(k,n-i-2));

	ret(i,0,n)Ans=(Ans+(S[i]^48)*A[n-i-1])%mod;

	printf("%lld\n",Ans);

}

signed main() {

	_main();

	return 0;

}