Divide by Zero 2021 and Codeforces Round #714 (Div. 2) E.Cost Equilibrium(思维题-网络流源汇思想)

题目

长度为n(n<=1e5)的数列a[](0<=ai<=1e9)，你每次可以选择i,j两个不同下标，

不妨ai<aj，你可以指定一个x，令aj-x，ai+x，然后代价是x*|j-i|，

如果对于一个a[]对应的排列(1 1 1 2 2这种排列只会被记一次方案)来说，

让所有数都相等的方案数的最小代价等于最大代价，

则称这种排列是好的，问有多少种排列是好的

思路来源

官方题解

题解

考虑如果和不能被n整除，输出0

否则计平均数为sum，小于sum的为汇，大于sum的为源，等于sum的为无关值

如果只有不超过一个源或者不超过一个汇，答案显然为可重排列数

不妨只有一个源，无论这个源在哪里，向各个地方流的方案都是唯一的

否则，源汇必须分离开，考虑2 4 4 2，交叉和就近显然对应了不同的答案，

所以，要么所有源都在所有汇前，要么所有汇都在所有源前，才能保证最小=最大

对其内部求可重排列数，最后的代价一定是把所有的源的代价都流向分界处，

再流向下一个值，期间能放就放，所以代价是唯一的，

注意到无关值的位置对答案是不影响的，所以如果有eq个无关值

那么先C(n,eq)一下，确定这些无关值的位置，

剩下的再考虑源汇的摆放方案对总方案乘积的贡献

代码

#include<iostream>
#include<cstdio>
#include<map>
#include<vector>
#include<algorithm>
using namespace std;
typedef long long ll;
const int mod=1e9+7,N=1e5+10;
int n,a[N],cnt,fac[N],finv[N],ans;
ll sum,eq;
vector<int>lb,ub,all;
int C(int n, int m){
	if(n<m || n<0 || m<0)return 0;
	return 1ll*fac[n]*finv[m]%mod*finv[n-m]%mod;
}
int cal(vector<int> &x){
	sort(x.begin(),x.end());
	int sz=x.size();
	int cnt=0,ans=fac[sz],lasv=-1;
	for(int i=0;i<sz;++i){
		if(x[i]!=lasv){
			ans=1ll*ans*finv[cnt]%mod;
			cnt=1;
		}
		else{
			cnt+=1;
		}
		lasv=x[i];
	}
	ans=1ll*ans*finv[cnt]%mod;
	return ans;
}
int main(){
	scanf("%d",&n);
	fac[0]=finv[0]=finv[1]=1;
	for(int i=1;i<=n;++i){
		if(i>=2)finv[i]=1ll*(mod-mod/i)*finv[mod%i]%mod;
		scanf("%d",&a[i]);
		sum+=a[i];
	}
	if(sum%n){
		puts("0");
		return 0;
	}
	sum/=n;
	for(int i=1;i<=n;++i){
		fac[i]=1ll*fac[i-1]*i%mod;
		finv[i]=1ll*finv[i-1]*finv[i]%mod;
		if(a[i]<sum)lb.push_back(a[i]);
		else if(a[i]>sum)ub.push_back(a[i]);
		else eq++;
		all.push_back(a[i]);
	}
	if((int)lb.size()<=1 || (int)ub.size()<=1){
		ans=cal(all);
	}
 	else{
        ans=2ll*cal(lb)*cal(ub)%mod*C(n,eq)%mod;
	}
	printf("%d\n",ans);
    return 0;
}
/*
4
0 4 0 4
*/