牛客练习赛42 C 反着计算贡献

https://ac.nowcoder.com/acm/contest/393/C

题意

给你一个矩阵, 每次从每行挑选一个数,组成一个排列,排列的和为不重复数字之和,求所有排列的和(n,m<=2000,a[i]<=1e9)

题解

• 假如计算数字x的贡献的话,x只会对所在排列最多贡献一次,那么只需要计算有多少排列含有x
• 反着计算每个数字的贡献,(总情况-包含这个数字的情况)*x

代码

#include<bits/stdc++.h>
#define ft first
#define se second
#define MOD 1000000007
#define MAXN 2005
#define mk make_pair
#define ll long long 
#define pii pair<ll,ll>
using namespace std;
ll pw(ll bs,ll x){
	ll ans=1;
	while(x){
		if(x&1)ans=ans*bs%MOD;
		bs=bs*bs%MOD;
		x>>=1;
	}
	return ans;
}
ll n,m,i,j,x,inv[MAXN],tol,C[MAXN],ans,nt,tp;
vector<pii>A;
int main(){
	scanf("%lld%lld",&m,&n);
	for(i=1;i<=n;i++)
		for(j=1;j<=m;j++){
			scanf("%lld",&x);
			A.push_back(mk(x,i));
		}
	inv[1]=1;for(i=2;i<MAXN;i++)inv[i]=inv[MOD%i]*(MOD-MOD/i)%MOD;
	sort(A.begin(),A.end());
	ll tol=pw(m,n);
	for(i=0;i<A.size();i=nt){
		tp=tol;
		for(nt=i;nt<A.size()&&A[i].ft==A[nt].ft;nt++){
			x=A[nt].se;
			tp=tp*inv[m-C[x]]%MOD;
			C[x]++;
			tp=tp*(m-C[x])%MOD;
		}
		for(j=i;j<nt;j++)C[A[j].se]=0;
		ans=(ans+(tol-tp+MOD)%MOD*A[i].ft%MOD)%MOD;
	}
	cout<<ans;
}