题目描述:
有N个线段,求任意组合能够形成三角形的情况概率
题目分析:
首先定义数组 num[i] 表示长度为i的线段个数
我们想要统计两个线段任意组合出长度为 i 的方案个数
可以先对 对 num 数组做一次FFT
然后减去同时选了自己和正反选择的情况就可以求出方案数了
对线段长度数组排序
我们枚举一个线段 假设它是三个里面最长的
根据三角形的定义,另外的两个线段的和要大于当前线段,可以通过对方案数数组求前缀得到
然后用容斥原理去除不合法的方案
1.一个大于当前线段,一个小于当前线段 i*(n-i-1)
2.与自己进行搭配 (n-1)
3.两个线段都大于当前线段 (n-i-1)*(n-i-2)/2
算出方案数 算一下 C(n,3) 就好了
题目链接:
Ac 代码:
#include <cstdio>
#include <iostream>
#include <cstring>
#include <cmath>
#include <algorithm>
#define int long long
const double PI=std::acos(-1);
const int maxm=2e6+100;
struct complex{
double real,imag;
complex(){};
complex(double _real,double _imag):real(_real),imag(_imag){}
};
inline complex operator + (complex x,complex y)
{
return (complex){x.real+y.real,x.imag+y.imag};
}
inline complex operator - (complex x,complex y)
{
return (complex){x.real-y.real,x.imag-y.imag};
}
inline complex operator * (complex x,complex y)
{
return (complex){x.real*y.real-x.imag*y.imag,x.real*y.imag+x.imag*y.real};
}
complex A[maxm];
int rev[maxm],len;
inline void FFT(complex *a,int n,int f)
{
for(int i=0;i<n;i++) if(i<rev[i]) std::swap(a[i],a[rev[i]]);
for(int i=1;i<n;i<<=1)
{
complex wn=(complex){std::cos(PI/i),f*std::sin(PI/i)};
for(int j=0;j<n;j+=(i<<1))
{
complex w=(complex){1,0};
for(int k=0;k<i;k++,w=(w*wn))
{
complex x=a[j+k],y=a[i+j+k]*w;
a[j+k]=x+y,a[i+j+k]=x-y;
}
}
}
if(f==-1) for(int i=0;i<n;i++) a[i].real/=n;
}
int sum[maxm],num[maxm],a[maxm];
int n,m;
void work()
{
memset(num,0,sizeof(num));
len=0;
scanf("%lld",&n);
for(int i=0,x;i<n;i++)
{
scanf("%d",&a[i]);
num[a[i]]++;
}
std::sort(a,a+n);
for(m=1;m<=2*(a[n-1]+1);m<<=1) len++;
for(int i=0;i<m;i++) rev[i]=((rev[i>>1]>>1)|((i&1)<<(len-1))),A[i]=(complex){0,0};
for(int i=0;i<=a[n-1];i++)
A[i]=(complex){num[i],0};
FFT(A,m,1);
for(int i=0;i<m;i++) A[i]=(A[i]*A[i]);
FFT(A,m,-1);
for(int i=0;i<m;i++)
num[i]=round(A[i].real);
m=2*a[n-1];
for(int i=0;i<n;i++)
num[a[i]+a[i]]--;
for(int i=0;i<=m;i++)
num[i]/=2;
for(int i=1;i<=m;i++)
sum[i]=sum[i-1]+num[i];
int ans=0;
for(int i=0;i<n;i++)
{
ans+=sum[m]-sum[a[i]];
ans-=(n-1-i)*i;
ans-=(n-1);
ans-=(n-1-i)*(n-i-2)/2;
}
int tot=n*(n-1)*(n-2)/6;
printf("%.7lf\n",(double)ans/tot);
}
signed main()
{
int T;
scanf("%lld",&T);
while(T--)
work();
return 0;
}