题意
给出一个长为n的序列,对于这个序列的任意一个排列,求 ∑ f a \sum f_a ∑fa的值。
题解
我们枚举每一个 a i a_i ai然后,计算 a i a_i ai出现了多少次,然后把 a i ∗ n u m a_i*num ai∗num加入到ans里面。
考虑 a i a_i ai什么时候出现。
a i a_i ai出现的位置前面的 a j a_j aj必须严格小于 a i a_i ai,这样 a i a_i ai才能被取到,而且 a i a_i ai后面一定要有比它大的数,这样才能 a i a_i ai取到。
假设小于 a i a_i ai的数有 m m m个,那么 n u m m = ∑ r = 0 m C m r ∗ r ! ∗ ( n − 1 − r ) ! num_m = \sum_{r = 0}^mC_m^r*r!*(n-1-r)! numm=∑r=0mCmr∗r!∗(n−1−r)!
整个的公式就是:
∑ i = 1 n a i ∗ n u m i \sum_{i = 1}^na_i*num_i ∑i=1nai∗numi
但这样暴力求的话时间复杂度是 O ( n 2 ) O(n^2) O(n2),因此必须运用公式变换上述式子,可以证明 n u m i num_i numi可以在 O ( 1 ) O(1) O(1)的时间求出来,具体求解方法见下图。
题解
#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
#define pr(x) cout<<#x<<":"<<x<<endl
const int maxn = 1e6+10;
typedef long long ll;
const ll mod = 1e9+7;
ll a[maxn],fac1[maxn],fac2[maxn];
int n;
int main(){
scanf("%d",&n);
for(int i = 0;i < n;++i)
scanf("%lld",&a[i]);
sort(a,a+n);
fac1[0] = 1;
fac2[n+1] = 1;
fac2[n] = n;
for(int i = 1;i <= n;++i)
fac1[i] = fac1[i-1]*i%mod;
for(int i = n-1;i >= 1;--i)
fac2[i] = fac2[i+1]*i%mod;
ll ans = 0;
for(int i = 0,j = 1;i < n && a[i] != a[n-1];i = j){
for(j = i;a[i] == a[j];j++);
ll num = j-i;
ans = (ans + num*((a[i]*fac1[n-i-1]%mod)*fac2[n-i+1]))%mod;
}
cout<<ans<<endl;
return 0;
}