[hackerrank Value of all Permutations]

题目大意

给定一个N个元素的序列A，有Q次询问，每次给出一个M，求对于A的所有排列跑下面的函数，返回值的和模P的答案

find(int permutation_A[], int M)
{
    x = Length(permutation_A)
    sum = 0
    for(i = 0; i < x; i++) {
        if (permutation_A[i] <= M)
            sum = sum + permutation_A[i]
        else
            break
    }
    return sum
}

N,Q≤$10^5$
A[i],M≤$10^9$
$10^6$≤P≤$2*10^6+3$且P为质数

分析

直接算答案很困难，可以从考虑每个元素对答案的贡献开始入手。
假设A中有k个数大于m，它们在排列中的位置是p1,p2…pk，那么对答案有贡献的只有n-k个数。考虑其中任意一个数A[i]，如果它在某个排列中对答案有贡献，那么一定对于任意所有j[1≤j≤k]，满足A[i] < <script type="math/tex" id="MathJax-Element-317"><</script>A[pj]
我们只用考虑这k+1个数的相对位置，假设共有P个排列，那么这个元素只在$\frac{P}{k+1}$个排列中有贡献。
所以对于一个M，在排序后的A中二分出一个位置，就能确定k了，然后剩下n-k个元素，用个前缀和即可。
剩下只有求P的大小了。假设一个数i在A中出现次数为ti，那么

$$P=\frac{N!}{\sum ti!}$$

时间复杂度$O((N+Q)logN)$

#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

const int N=100005;

typedef long long LL;

int n,q,m,mo,a[N],Fac[N],Inv[N],Fac_Inv[N],ans,s[N],l,r,mid,P;

char c;

int read()
{
    for (c=getchar();c<'0' || c>'9';c=getchar());
    int x=c-48;
    for (c=getchar();c>='0' && c<='9';c=getchar()) x=x*10+c-48;
    return x;
}

int main()
{
    mo=read();
    n=read();
    for (int i=1;i<=n;i++) a[i]=read();
    sort(a+1,a+n+1);
    Fac[0]=Fac[1]=Fac_Inv[0]=Fac_Inv[1]=Inv[1]=1;
    for (int i=2;i<=n;i++)
    {
        Fac[i]=(LL)Fac[i-1]*i%mo;
        Inv[i]=(LL)Inv[mo%i]*(mo-mo/i)%mo;
        Fac_Inv[i]=(LL)Fac_Inv[i-1]*Inv[i]%mo;
    }
    P=Fac[n];
    for (int i=1;i<=n;i++) if (i==n || a[i]!=a[i+1])
    {
        int j=0;
        for (;i!=j && a[i]==a[i-j];j++);
        P=(LL)P*Fac_Inv[j]%mo;
    }
    q=read();
    for (int i=1;i<=n;i++) s[i]=(s[i-1]+a[i])%mo;
    while (q--)
    {
        m=read();
        for (l=1,r=n+1,mid=l+r>>1;l<r;mid=l+r>>1)
            if (a[mid]<=m) l=mid+1;else r=mid;
        l--;
        ans=(LL)s[l]*P%mo*Inv[n-l+1]%mo;
        printf("%d\n",ans);
    }
    return 0;
}