Description
有两个队伍A和B,每个队伍都有n个人。这两支队伍之间进行n场1对1比赛,每一场都是由A中的一个选手与B中的一个选手对抗。同一个人不会参加多场比赛,每个人的对手都是随机而等概率的。例如A队有A1和A2两个人,B队有B1和B2两个人,那么(A1 vs B1,A2 vs B2)和(A1 vs B2,A2 vs B1)的概率都是均等的50%。
每个选手都有一个非负的实力值。如果实力值为X和Y的选手对抗,那么实力值较强的选手所在的队伍将会获得(X-Y)^2的得分。
求A的得分减B的得分的期望值。
Input
第一行一个数n表示两队的人数为n。
第二行n个数,第i个数A[i]表示队伍A的第i个人的实力值。
第三行n个数,第i个数B[i]表示队伍B的第i个人的实力值。
Output
输出仅包含一个实数表示A期望赢B多少分。答案保留到小数点后一位(注意精度)。
Hint
对于30%的数据,n≤50。
对于100%的.据,n≤50000;A[i],B[i]≤50000。
Analysis
题目所求的是
∑ni=1∑nj=1±(a[i]−b[j])2n
其中正负由 a[i] 和 b[i] 的大小关系决定分值的正负
上式展开后可得
a[i]2+b[j]2+2∗a[i]∗b[j]
对于每个 a[i] 可以预处理 b 的前缀和与前缀平方和
正负的处理可以对
精度坑人,输给%.1Lf
记住了
Code
#include <algorithm>
#include <stdio.h>
using namespace std;
long long n,sum[50001],a[50001],b[50001];
long double qrt[50001];
int main()
{
scanf("%lld",&n);
sum[0]=0;qrt[0]=0;
long double m=n*1.0,ans=0.0;
for (int i=1;i<=n;i++)
scanf("%lld",&a[i]);
for (int i=1;i<=n;i++)
scanf("%lld",&b[i]);
sort(a+1,a+n+1);
sort(b+1,b+n+1);
for (int i=1;i<=n;i++)
sum[i]=sum[i-1]+b[i],
qrt[i]=qrt[i-1]+b[i]*b[i];
int t=0;
for (int i=1;i<=n;i++)
{
while (t<n&&a[i]>b[t+1])t++;
ans+=(t*a[i]*a[i]+qrt[t]-2*a[i]*sum[t])/m;
ans-=((n-t)*a[i]*a[i]+qrt[n]-qrt[t]-2*a[i]*(sum[n]-sum[t]))/m;
}
printf("%.1Lf\n",ans);
return 0;
}