CodeForces 520E. Pluses everywhere 字符串,组合数学

CodeForces 520E. Pluses everywhere 字符串,组合数学_第1张图片CodeForces 520E. Pluses everywhere 字符串,组合数学_第2张图片

题意:给定长度为n的数字串,在空隙添上k个+号,可以得到一个式子,求出插入k个+号可以得到的所有式子的计算结果之和

公式题:

代码中a[i]与上面公式中是反的,代码中a[i]是从左往右的。

在对M=10^9 +7 取模的情况下,计算组合数C(n,k)可以预处理fac[i](i 的阶乘)和 revfac[i](i 在模M下的乘法逆)

然后线性递推一下系数数组,最后求和。

要说明的是,代码中的C(n,k)没有处理n<0而k>=0的情况(这时不一定为0,而代码中直接令它为0)

所以上述公式直接使用会有误差,于是代码中特判了k=0的情况,避免组合数出问题。



代码:

#include <iostream>
#include <cstdio>
#include <cmath>
#define maxn 100007
#define M 1000000007
#define LL long long
using namespace std;
LL fac[maxn];
LL ten[maxn];
LL revfac[maxn];
LL QuickPower(LL a,LL k){//快速幂
	LL ANS=1;
	while(k){
		if(k&1) ANS=(ANS*a)%M;
		k>>=1;
		a=(a*a)%M;
	}
	return ANS;
}
LL C(LL n,LL k){//组合数
	if(k<0||k>n ||n <0) return 0;
	LL ANS=fac[n];
	ANS=(ANS*revfac[k])%M;
	ANS=(ANS*revfac[n-k])%M;
	return ANS;
}
void Init(){//初始化阶乘fac数组以及阶乘乘法逆revfac数组
	ten[0]=fac[0]=revfac[0]=1;
	for(int i=1;i<maxn;++i){
		fac[i]=(i*fac[i-1])%M;
		revfac[i]=QuickPower(fac[i],M-2);
		ten[i]=(10*ten[i-1])%M;
	}
}
int n,k;
char str[maxn];
LL Coef[maxn];
int main(void)
{
	Init();
	while(~scanf("%d%d",&n,&k)){
		scanf("%s",str);
		if(k==0){
			LL ANS=0;
			for(int i=0;i<n;++i) ANS=(ANS+ten[i]*(str[n-i-1]-'0'))% M;
			printf("%d\n",(int)ANS);
			continue;
		}
		int D=n-k-1;//计算系数
		Coef[0]=C(n-2,k-1);
		for(int i=1;i<=D;++i) Coef[i]=(Coef[i-1]+ten[i]*C(n-i-2,k-1))% M;
		for(int i=D+1;i<n;++i) Coef[i]=Coef[i-1];
		for(int i=0;i<=D;++i) Coef[i]=(Coef[i]+ten[i]*C(n-i-2,k))% M;
		
		LL ANS=0;//计算最终答案
		for(int i=0;i<n;++i){
			ANS=(ANS+Coef[n-i-1]*(str[i]-'0'))% M;
		}
		printf("%d\n",(int)ANS);
	}
return 0;
}



你可能感兴趣的:(字符串,codeforces)