CodeForces 340 C.Tourist Problem（组合数学）

Description

给出数轴上$n$个地点的坐标$a_1,...,a_n$，随机的给出一个排列，按照该排列去一个个到达排列对应的地点，初始在原点处，问所走距离的期望值

Input

第一行一整数$n$，之后输入$n$个整数$a_i(2\le n\le 10^5,1\le a_i\le 10^7)$

Output

输出距离期望值的最简分数表示

Sample Input

3
2 3 5

Sample Output

22 3

Solution

对于一个排列$p_1,...,p_n$，其对应的距离为$a_{p_1}+\sum\limits_{i=1}^{n-1}|a_{p_i}-a_{p_{i+1}}|$

考虑$a_i$对答案的影响（假设$a$序列已经升序）

如果$p_1=i$，考虑$p_2$的取值，其他$n-2$个数字随便排，如果$p_2>p_1$，那么一加一减，$a_i$对答案贡献是$0$，如果$p_2<p_1$，那么两次都是加，对答案贡献是$2a_i$，故这种情况对答案的贡献就是$2\cdot(i-1)\cdot(n-2)!\cdot a_i$

如果$p_n=i$，考虑$p_{n-1}$的取值，其他$n-2$个数字随便排，如果$p_{n-1}>p_n$，那么对答案贡献是$-a_i$，如果$p_{n-1}<p_n$，那么对答案贡献是$a_i$，故这种情况对答案的贡献就是$((i-1)-(n-i))\cdot (n-2)!\cdot a_i$

如果$p_j=i,2\le j\le n-1$，考虑$p_{j-1}$和$p_{j+1}$的取值，其他$n-3$个数字随便排，如果两者比$p_j$一大一小，则对答案贡献是$0$，如果都比$p_j$大，那么对答案贡献就是$-2\cdot a_i$，如果都比$p_j$小，那么对答案贡献就是$2\cdot a_i$，故这种情况对答案的贡献就是$(n-2)\cdot (n-3)!\cdot 2\cdot (A_{i-1}^2-A_{n-i}^2)\cdot a_i$

故$a_i$对总距离和贡献的次数为$(n-2)!\cdot (2\cdot(i-1)+2\cdot i-n-1+2\cdot (A_{i-1}^2-A_{n-i}^2))=(n-1)!\cdot(4\cdot i-2\cdot n-1)$

进而求得期望为$\frac{\sum\limits_{i=1}^n(4i-2n-1)\cdot a_i}{n}$

Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
typedef pair<int,int>P;
const int INF=0x3f3f3f3f,maxn=100005;
ll gcd(ll a,ll b)
{
    return b?gcd(b,a%b):a;
}
int n,a[maxn];
int main()
{
    int n;
    while(~scanf("%d",&n))
    {
        for(int i=1;i<=n;i++)scanf("%d",&a[i]);
        sort(a+1,a+n+1);
        ll ans=0;
        for(int i=1;i<=n;i++)
            ans+=(ll)(4*i-2*n-1)*a[i];
        ll g=gcd(ans,n);
        printf("%I64d %I64d\n",ans/g,n/g);
    }
    return 0;
}