[HDU 4609] 3-idiots

题目描述：

有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) 就好了

题目链接：

HDU 4609
BZOJ 3513

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;                 
}