CodeForces - 626D
前缀和+概率
cnt[i]表示差为i的数对的数量;sum[]为cnt的前缀和
i表示第一回合的差,j表示第二回合的差,i+j小于第3回合,第三回合一共sum[5000]-sun[i+j],最后用组合公式算一下
#include <iostream> #include <cstdio> #include <queue> #include <algorithm> #include <cmath> #include <cstring> #define inf 2147483647 #define N 1000010 #define p(a) putchar(a) #define For(i,a,b) for(int i=a;i<=b;++i) using namespace std; int n; double a[N],sum[N],cnt[N],ans,t; void in(int &x){ int y=1;char c=getchar();x=0; while(c<'0'||c>'9'){if(c=='-')y=-1;c=getchar();} while(c<='9'&&c>='0'){ x=(x<<1)+(x<<3)+c-'0';c=getchar();} x*=y; } void o(int x){ if(x<0){p('-');x=-x;} if(x>9)o(x/10); p(x%10+'0'); } signed main(){ in(n); For(i,1,n) cin>>a[i]; sort(a+1,a+n+1); For(i,2,n) For(j,1,i-1) cnt[int(a[i]-a[j])]=cnt[int(a[i]-a[j])]+1.0; For(i,1,5000) sum[i]=sum[i-1]+cnt[i]; t=n*(n-1)/2; For(i,1,5000) For(j,1,5000){ if(i+j<5000){ ans+=cnt[i]/t*cnt[j]/t*(sum[5000]-sum[i+j])/t; } } printf("%.10f ",ans); return 0; }