hdu5160 Apple 组合数学

问题描述

现在要给n个小朋友发苹果，每一个小朋友都会有自己相应的苹果数目 {A_1},{A_2},{A_3}, \cdots {A_n}A​1​​,A​2​​,A​3​​,⋯A​n​​。  发苹果的时候，小朋友先从左到右随机站成一排。设站好后从左到右小朋友的标号是 {p_1},{p_2},{p_3}, \cdots {p_n}p​1​​,p​2​​,p​3​​,⋯p​n​​，他们所需要的苹果数目是 {A_{{p_1}}},{A_{{p_2}}},{A_{{p_3}}}, \cdots {A_{{p_n}}}A​p​1​​​​,A​p​2​​​​,A​p​3​​​​,⋯A​p​n​​​​，对于最左边的人要给他发 {A_{{p_1}}}A​p​1​​​​个苹果，对于第i (i>1)个人，如果前面有一个人的苹果要求的数目比他多，那么就不用发给他苹果，否则发给他 {A_{{p_i}}}A​p​i​​​​个苹果。那么对于某一个排列需要多少苹果就可以计算出来了。现在我们想知道对于所有的排列总共要发多少苹果。两个排列不一样当且仅当至少存在一个位置要求的苹果数目不一样。

题意即一个序列的所有不同排列的总和，一个序列的和如上所述。

分开计算每个值得贡献，如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;
}