问题描述
现在要给n个小朋友发苹果,每一个小朋友都会有自己相应的苹果数目 A1,A2,A3,⋯An。 发苹果的时候,小朋友先从左到右随机站成一排。设站好后从左到右小朋友的标号是 p1,p2,p3,⋯pn,他们所需要的苹果数目是 Ap1,Ap2,Ap3,⋯Apn,对于最左边的人要给他发 Ap1个苹果,对于第i (i>1)个人,如果前面有一个人的苹果要求的数目比他多,那么就不用发给他苹果,否则发给他 Api个苹果。那么对于某一个排列需要多少苹果就可以计算出来了。现在我们想知道对于所有的排列总共要发多少苹果。两个排列不一样当且仅当至少存在一个位置要求的苹果数目不一样。
题意即一个序列的所有不同排列的总和,一个序列的和如上所述。
分开计算每个值得贡献,如j个i的所有方案数的总和。那里乘C(n,n-sum[i-1])非常妙。
#include<iostream>
#include<cstring>
#include<cstdio>
#include<ostream>
#include<istream>
#include<algorithm>
#include<queue>
#include<string>
#include<cmath>
#include<set>
#include<map>
#include<stack>
#include<vector>
#define fi first
#define se second
#define ll long long
#define pii pair<int,int>
#define inf (1<<30)
#define eps 1e-8
#define pb push_back
using namespace std;
const int maxn=110005;
const ll mod=1000000007;
ll jiech[maxn];
ll inv[maxn];
int n;
int d[maxn];
int dis[maxn];
int num[maxn];
int p[maxn];
int sum[maxn];
ll tot[maxn];
ll tui[maxn];
ll powMod(ll a,ll b)
{
if(b==0) return 1;
ll ans=powMod(a,b/2);
ans=ans*ans%mod;
if(b&1) ans=ans*a%mod;
return ans;
}
void pre()
{
jiech[0]=1;
int N=100000;
for(int i=1;i<=N;i++)
jiech[i]=jiech[i-1]*i%mod;
inv[N]=powMod(jiech[N],mod-2);
for(int i=N-1;i>=0;i--)
inv[i]=inv[i+1]*(i+1)%mod;
}
ll C(int a,int b)
{
return jiech[a]*inv[b]%mod*inv[a-b]%mod;
}
int main()
{
pre();
int t;
int cas=1;
scanf("%d",&t);
while(t--) {
scanf("%d",&n);
for(int i=1;i<=n;i++) {
num[i]=0;
scanf("%d",&d[i]);
dis[i-1]=d[i];
}
sort(dis,dis+n);
int k=unique(dis,dis+n)-dis;
for(int i=1;i<=n;i++) {
int u=d[i];
d[i]=lower_bound(dis,dis+k,d[i])-dis+1;
p[d[i]]=u;
num[d[i]]++;
}
sum[0]=0;
for(int i=1;i<=k;i++)
sum[i]=sum[i-1]+num[i];
tot[0]=1;
for(int i=1;i<=k;i++)
tot[i]=tot[i-1]*inv[num[i]]%mod;
tui[k+1]=1;
for(int i=k;i>=1;i--)
tui[i]=tui[i+1]*inv[num[i]]%mod;
ll ans=0;
for(int i=1;i<=k;i++) {
for(int j=1;j<=num[i];j++) {
//j个i的贡献
ll u=(jiech[n-sum[i-1]-j]*tui[i+1]%mod*inv[num[i]-j]%mod-jiech[n-sum[i-1]-j-1]*tui[i+1]%mod*inv[num[i]-j-1]%mod+mod)%mod;
u=u*C(n,n-sum[i-1])%mod*jiech[sum[i-1]]%mod*tot[i-1]%mod;
u=u*j%mod*p[i]%mod;
ans=(ans+u)%mod;
}
}
printf("Case #%d: %I64d\n",cas++,ans);
}
return 0;
}