LightOJ - 1213(组合数学 快速幂)

https://vjudge.net/contest/276155#problem/L

给出n,k,mod 和数量为n得int数组a[i],求k层嵌套得a数组相加后膜mod的值。

排列组合求得每个数的出现次数为(n^(k-1))*k,再用快速幂求结果。

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
using namespace std;

long long kuai_mi(int n,int k,int mod){
	long long a = 1;
	while(k){
		if(k&1) a = (a*n)%mod;
		n = (n*n)%mod;
		k >>= 1;
	}
	return a;
}

int main(){
	int n,k,mod,x,t,id = 0;
	long long ans,p;
	scanf("%d",&t);
	while(t--){
		scanf("%d%d%d",&n,&k,&mod);
		ans = 0;
		for(int i = 0;i<n;++i){
			scanf("%d",&x);
			ans = (ans+x%mod)%mod;
		}
		p = kuai_mi(n,k-1,mod);
		ans = (p*(ans * (k%mod))%mod)%mod;
		printf("Case %d: ",++id);
		cout<<ans<<endl;
	}
	return 0;
}

猜你喜欢

转载自blog.csdn.net/dukig/article/details/89458730