[TJOI2018]教科书般的亵渎

Description
小豆喜欢玩游戏, 现在他在玩一个游戏遇到这样的场面,每个怪的血量为ai,且每个怪物血量均不相同, 小豆手里有无限张“亵渎”。亵渎的效果是对所有的怪造成1点伤害,如果有怪死亡,则再次施放该法术。我们认为血量为0怪物死亡。小豆使用一张“亵渎”会获得一定的分数,分数计算如下,在使用一张“亵渎”之后,每一个被亵渎造成伤害的怪会产生x^k,其中x是造成伤害前怪的血量为x和需要杀死所有怪物所需的“亵渎”的张数k。

Input
第一行输入一个T,表示有多少组测试数据。
每组组测试数据第一行为n,m,表示有当前怪物最高的血量n,和m种没有出现的血量。
接下来m行,每行1个数ai,表示场上没有血量为ai的怪物。
T ≤10 n ≤ 10^13 m ≤ 50

Output
一共T行,每行一个数,第i行表示第i组测试数据中小豆的最后可以获得的分数
因为这个分数会很大,需要模10^9 + 7。

Sample Input
2
10 1
5
4 2
1
2

Sample Output
415
135


MD……看题看半天系列

\(k\)一直是固定的,为\(m+1\)

每次得分是\(\sum\limits_{i=1}^{n}i^k-\sum\limits_{j=1}^ma_j^k\),然后\(n=n-a_1\)\(a_1\)消失,之后的所有\(a_j\)依次向前挪

所以……这题瓶颈在求幂和,求幂和参考这篇博客即可

/*program from Wolfycz*/
#include<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define inf 0x7f7f7f7f
typedef long long ll;
typedef unsigned int ui;
typedef unsigned long long ull;
inline char gc(){
	static char buf[1000000],*p1=buf,*p2=buf;
	return p1==p2&&(p2=(p1=buf)+fread(buf,1,1000000,stdin),p1==p2)?EOF:*p1++;
}
template<typename T>inline T frd(T x){
	int f=1; char ch=gc();
	for (;ch<'0'||ch>'9';ch=gc())	if (ch=='-')    f=-1;
	for (;ch>='0'&&ch<='9';ch=gc())	x=(x<<1)+(x<<3)+ch-'0';
	return x*f;
}
template<typename T>inline T read(T x){
	int f=1;char ch=getchar();
	for (;ch<'0'||ch>'9';ch=getchar())	if (ch=='-')	f=-1;
	for (;ch>='0'&&ch<='9';ch=getchar())	x=(x<<1)+(x<<3)+ch-'0';
	return x*f;
}
inline void print(int x){
	if (x<0)    putchar('-'),x=-x;
	if (x>9)	print(x/10);
	putchar(x%10+'0');
}
template<typename T>inline T min(T x,T y){return x<y?x:y;}
template<typename T>inline T max(T x,T y){return x>y?x:y;}
const int N=50,p=1e9+7;
int prime[N+10],f[N+10],fac[N+10],inv[N+10];
bool inprime[N+10];
int mlt(int a,int b){
	int res=1;
	for (;b;b>>=1,a=1ll*a*a%p)	if (b&1)	res=1ll*res*a%p;
	return res;
}
void prepare(int k){
	int tot=0; f[1]=fac[0]=inv[0]=inv[1]=1,k++;
	for (int i=2;i<=k;i++){
		if (!inprime[i])	prime[++tot]=i,f[i]=mlt(i,k-1);
		for (int j=1;j<=tot&&i*prime[j]<=k;j++){
			inprime[i*prime[j]]=1;
			f[i*prime[j]]=1ll*f[i]*f[prime[j]]%p;
			if (i%prime[j]==0)	break;
		}
	}
	for (int i=1;i<=k;i++)	f[i]=(f[i]+f[i-1])%p;
	
	for (int i=1;i<=k;i++)	fac[i]=1ll*fac[i-1]*i%p;
	for (int i=2;i<=k;i++)	inv[i]=1ll*(p-p/i)*inv[p%i]%p;
	for (int i=1;i<=k;i++)	inv[i]=1ll*inv[i-1]*inv[i]%p;
}
int calc(int k,ll n){
	static int Pre[N+10],Suf[N+10];
	n%=p,k++,Pre[0]=n,Suf[k]=n-k; int res=0;
	for (int i=1;i<=k;i++)	Pre[i]=1ll*Pre[i-1]*(n-i)%p;
	for (int i=k-1;~i;i--)	Suf[i]=1ll*Suf[i+1]*(n-i)%p;
	for (int i=0;i<=k;i++){
		int tmp=1ll*f[i]*inv[i]%p*inv[k-i]%p;
		if (i!=0)	tmp=1ll*tmp*Pre[i-1]%p;
		if (i!=k)	tmp=1ll*tmp*Suf[i+1]%p;
		res=(res+((k-i)&1?-1:1)*tmp)%p;
	}
	return (res+p)%p;
}
int main(){
	for (int T=read(0);T;T--){
		static ll A[N+10];
		ll n=read(0ll); int m=read(0),k=m+1;
		for (int i=1;i<=m;i++)	A[i]=read(0ll);
		std::sort(A+1,A+1+m);
		prepare(k); int Ans=0;
		for (int i=1;i<=m;i++){
			Ans=(Ans+calc(k,n))%p;
			for (int j=i+1;j<=m;j++)	Ans=(Ans-mlt(A[j],k))%p,A[j]-=A[i];
			Ans=(Ans-mlt(A[i],k)),n-=A[i];
		}
		Ans=(Ans+calc(k,n))%p;
		printf("%d\n",(Ans+p)%p);
	}
	return 0;
}
posted @ 2019-03-31 09:27  Wolfycz  阅读(273)  评论(0编辑  收藏  举报