jzoj5945

最新推荐文章于 2023-12-23 22:12:34 发布

weixin_30321449

最新推荐文章于 2023-12-23 22:12:34 发布

阅读量117

点赞数

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/rilisoft/p/10385229.html

這題是均分紙牌求方案數
我們可以分2種情況討論：
1.當前面部分的平均值>=現在我們要的值
那麼我們可以把所有牌都放到第i個點，現在若有k張牌，則要把 $k - i * a v e$ 張牌放到後面
對答案的貢獻為 $c (a [i], k - i * a v e)$ ，要更新現在的元素值
2.當前面部分平均值<現在我們要的值
那麼我們可以從後面抓 $i * a v e - k$ 張牌
由於我們要讓所有部分有 $a v e$ 張牌，所以我們後面需要有 $a v e + i * a v e - k$ 張牌，才可以讓抓牌後這部分有 $a v e$ 張牌
所以對答案的貢獻為 $c ((i + 1) * a v e - s [i], i * a v e - s [i])$

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
#define mo 998244353ll
ll t,n,a[1010],jc[1000010],ijc[1000010],s[1010];
ll qp(ll x,ll y){
	ll r=1;
	while(y){
		if(y&1)r=r*x%mo;
		x=x*x%mo;
		y>>=1;
	}
	return r;
}
void get(){
	jc[0]=1;
	for(ll i=1;i<=1000005;i++)
		jc[i]=jc[i-1]*i%mo;
	ijc[0]=1;
	for(ll i=1;i<=1000005;i++)
		ijc[i]=ijc[i-1]*qp(i,mo-2)%mo; 
}
ll c(ll x,ll y){
	return jc[x]*ijc[x-y]%mo*ijc[y]%mo;
}
int main(){
	freopen("gwent.in","r",stdin);
	freopen("gwent.out","w",stdout);
	scanf("%lld",&t);
	get();
	while(t--){
		scanf("%lld",&n);
		for(ll i=1;i<=n;i++)
			scanf("%lld",&a[i]);
		for(ll i=1;i<=n;i++)
			s[i]=s[i-1]+a[i];
		ll k=s[n]/n,ans=1;
		for(ll i=1;i<=n;i++){
			ll va=s[i]-k*i;
			if(va>=0){
				ans=ans*c(a[i],va)%mo;
				a[i]-=va;
				a[i+1]+=va;
			}
			else ans=ans*c(-va+k,-va)%mo;
		}
		printf("%lld\n",ans);
	}
}