HDU 4609 3-idiots（FFT）

题意：给出n个正整数（数组A）。每次随机选出三个数。问这三个数能组成三角形的概率为多大？

首先，我们用类似桶排计数的方法作出两个数组a,b，储存每个长度有几条边，然后对两个数组求卷积。

求出卷积后，这就代表了2条边能构成的边长度的集合了，注意，由于求卷积的时候可能把两条相同的边相加，所以最后求出的数组一定要减去这重复的部分，然后，先算x后算y等价于先算y后算x，所以要除以二。

然后，对于第三条边a[i]，我们这样考虑：令它作为这三条边中最大的那条！

所以之前的卷积求出来的两边和一定会比这条边大，所以每次都计数：ans+=sum[mx]-sum[a[i]]；

然后减去重复部分：

    （1）一个选了i后面，一个选了i前面 （减去（i-1)*(n-i)个）

    （2）一个选了i，另一个随意 （减去n-1个）

    （3）两个都是i后面的 （减去C（n-i,2)个)；

非常interesting的题目！

#include<algorithm>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<iostream>
#include<complex>
#define N 300005
#define ll long long
typedef std::complex<double> cd;
int a[400005];
ll C[400005],Sum[400005],A[400005],B[400005];
int n,mxnum;
int bitrev(int t,int n){
    int res=0;
    for (int i=0;i<n;i++) res|=((t>>(n-i-1))&1)<<i;//括号要多加 
    return res;
}
void fft(cd *a,int n,int rev){
    int len=1<<n;
    static cd y[N];double Pi=acos(-1);
    for (int i=0;i<len;i++) y[i]=a[bitrev(i,n)];
    for (int d=1;d<len;d<<=1){
        cd wn(exp(cd(0,Pi*rev/d)));
        for (int k=0;k<len;k+=2*d){
            cd w(1,0);
            for (int i=k;i<k+d;i++,w*=wn){
                cd u=y[i],v=w*y[i+d];
                y[i]=u+v;
                y[i+d]=u-v;
            }
        }
    }
    if (rev==-1)
    for (int i=0;i<len;i++) y[i]/=len;
    for (int i=0;i<len;i++) a[i]=y[i];
}
void mul(ll *a,int la,ll *b,int lb,ll *c,int &lc){
    int len=1,n=0;
    static cd t1[N],t2[N];
    for (;len<la*2||len<lb*2;len<<=1,n++);
    for (int i=0;i<la;i++) t1[i]=cd(a[i],0);
    for (int i=0;i<lb;i++) t2[i]=cd(b[i],0);
    for (int i=la;i<len;i++) t1[i]=cd(0,0);
    for (int i=lb;i<len;i++) t2[i]=cd(0,0);
    fft(t1,n,1);fft(t2,n,1);
    for (int i=0;i<len;i++) t1[i]*=t2[i];
    fft(t1,n,-1);
    for (int i=0;i<len;i++) c[i]=(long long)(t1[i].real()+0.5);
    lc=len;
}
int main(){
    int T;
    scanf("%d",&T);
    while (T--){
        scanf("%d",&n);
        for (int i=1;i<=n;i++) scanf("%d",&a[i]);
        mxnum=0;
        for (int i=1;i<=n;i++) mxnum=std::max(mxnum,a[i]);
        for (int i=1;i<=n;i++) A[a[i]]++,B[a[i]]++;
        int lc;
        mul(A,mxnum+1,B,mxnum+1,C,lc);
        for (int i=1;i<=n;i++) C[a[i]*2]--;
        for (int i=1;i<=mxnum*2;i++) C[i]/=2;
        Sum[0]=0;
        for (int i=1;i<=mxnum*2;i++) 
         Sum[i]=Sum[i-1]+C[i];
        std::sort(a+1,a+1+n);
        ll ans=0;
        for (int i=1;i<=n;i++){
            ans+=Sum[mxnum*2]-Sum[a[i]];
            ans-=(ll)n-1;
            ans-=(ll)(i-1)*(n-i);
            ans-=(ll)(n-i)*(n-i-1)/2;
        }
        ll sum=(ll)(n)*(n-1)*(n-2)/6;
        printf("%.7f
",(double)1.0*((double)(ans))/((double)sum));
        for (int i=0;i<=mxnum;i++) A[i]=B[i]=Sum[i]=C[i]=0;
    }
}