CodeForces 520E Pluses everywhere

链接:http://codeforces.com/problemset/problem/520/E

组合数+按位考虑,从第一位往后枚举,考虑第i位做个位、十位、百位。。的贡献,这样显然每个数的总贡献就是答案,而考虑个位十位百位那一步是可以预处理出一个组合数前缀和C(n-2,k-1),C(n-3,k-1)...然后算某一位就是O(1)的了。。

#pragma GCC optimize(3,"inline","Ofast")
#include<bits/stdc++.h>
#define ll long long
using namespace std;
const ll mod=1e9+7;
int n,k,a[100010];char s[100010];
ll mo[100010],sum[100010],jc[100010],ss[100010];
ll qpow(ll x,ll y)
{
	ll res=1;
	while(y)
	{
		if(y&1)res=res*x%mod;
		x=x*x%mod,y>>=1;
	}
	return res;
}
ll C(ll n,ll m)
{return jc[n]*qpow(jc[n-m]*jc[m]%mod,mod-2)%mod;}
int main()
{
	ll ans=0,tp=0,nw=0;
	scanf("%d%d",&n,&k);
	scanf("%s",s+1);
	jc[0]=1;
	for(int i=1;i<=n;i++)
		jc[i]=jc[i-1]*i%mod;
	for(int i=1;i<=n;i++)
		a[i]=s[i]-'0';
	mo[0]=1;
	for(int i=1;i<=n;i++)
		mo[i]=mo[i-1]*10%mod;
	for(int i=1;i+k-1<=n-1;i++)
		ss[i]=ss[i-1],ss[i]=(ss[i]+mo[i-1]*C(n-1-i,k-1)%mod)%mod;
	for(int i=1;i<=n;i++)
	{
		tp=n-i,tp=min(tp,(ll)n-k);
		ans=(ans+ss[tp]*a[i]%mod)%mod;
		if(i-1>=k)ans=(ans+mo[n-i]*a[i]%mod*C(i-1,k)%mod)%mod;
	}
	cout<<ans;
}

猜你喜欢

转载自blog.csdn.net/caoyang1123/article/details/81513016
今日推荐