题意:给定 n 个正整数的数组,求从中随机选取三个数 ai , aj , ak 可以构成三角形的概率。
分析:纯暴力枚举肯定不行,得用到FFT优化(不明白的可以看一下:https://blog.csdn.net/ZLH_HHHH/article/details/75604333,讲得蛮不错的)
总方案数:
设满足条件的方案数位 ans,则 ans / all 就是所求概率,所以主体就是求 ans 的过程;
具体一些地方我跟着代码注释,会比较容易理解。
代码:
#include<cstdio>
#include<cstring>
#include<complex>
#include<iostream>
#include<algorithm>
#include<functional>
using namespace std;
typedef long long ll;
const double PI=acos(-1.0);
struct Complex
{
double r,i;
Complex(double _r=0,double _i=0)
{
r=_r,i=_i;
}
Complex operator + (const Complex &b)
{
return Complex(r+b.r,i+b.i);
}
Complex operator - (const Complex &b)
{
return Complex(r-b.r,i-b.i);
}
Complex operator * (const Complex &b)
{
return Complex(r*b.r-i*b.i,r*b.i+i*b.r);
}
};
void change(Complex y[],int len)
{
int i,j,k;
for(i=1,j=len/2;i<len-1;i++)
{
if(i<j) swap(y[i],y[j]);
k=len/2;
while(j>=k)
{
j-=k;
k/=2;
}
if(j<k) j+=k;
}
}
void fft(Complex y[],int len,int on)
{
change(y,len);
for(int h=2;h<=len;h<<=1)
{
Complex wn(cos(-on*2*PI/h),sin(-on*2*PI/h));
for(int j=0;j<len;j+=h)
{
Complex w(1,0);
for(int k=j;k< j+h/2;k++)
{
Complex u=y[k];
Complex t=w*y[k+h/2];
y[k]=u+t;
y[k+h/2]=u-t;
w=w*wn;
}
}
}
if(on==-1)
for(int i=0;i<len;i++)
y[i].r/=len;
}
const int N = 1e5+10;
Complex x1[N*4];
int a[N];
ll num[N*4];
ll sum[N*4];
int main()
{
int T;
scanf("%d",&T);
while(T--)
{
memset(num,0,sizeof(num));
int n;
scanf("%d",&n);
for(int i=0;i<n;i++) scanf("%d",&a[i]),num[a[i]]++;
sort(a,a+n);
int len=1,maxa=a[n-1]+1;
while(len<maxa*2) len<<=1;
for(int i=0;i<maxa;i++)
x1[i]=Complex(num[i],0);
for(int i=maxa;i<len;i++)
x1[i]=Complex(0,0);
fft(x1,len,1);
for(int i=0;i<len;i++) x1[i]=x1[i]*x1[i];
fft(x1,len,-1);
for(int i=0;i<len;i++)
num[i]=(ll)(x1[i].r+0.5);
/* 以上都是FFT 的常规操作,就不解释了 */
len=a[n-1]*2;
//小优化,枚举数组内两个数的和最大就是a[n-1]*2了
for(int i=0;i<n;i++) num[a[i]+a[i]]--;
//因为一个数只能取一次,所以除去重复的情况
for(int i=0;i<=len;i++) num[i]/=2;
//取i,j 和 j,i 同效,故除以2
sum[0]=0;
for(int i=1;i<=len;i++) sum[i]=sum[i-1]+num[i];
ll ans=0;
for(int i=0;i<n;i++)
{
ans+=sum[len]-sum[a[i]];
//一:以当前a[i]为三角形最长的边,
// 统计其余两边之和大于这一边的方案数
ans-=(ll)(n-i-1)*(n-i-2)/2;
//减去一中两边之和来源于两个比a[i]大的边组合的情况
ans-=(ll)(n-i-1)*i;
//减去一中两边之和来源于一个比a[i]大,一个比a[i]小的情况
ans-=(n-1);
//减去一中两边之和有一边来源于a[i]的重复情况
}
ll all=(ll)n*(n-1)*(n-2)/6;
printf("%.7lf\n",(double)ans/all);
}
return 0;
}
/*
2
4
1 3 3 4
4
2 3 3 4
*/