HDU 4709 3-idiots FFT 多项式

http://acm.hdu.edu.cn/showproblem.php?pid=4609

给一堆边，求这一堆边随便挑三个能组成三角形的概率。

裸fft，被垃圾题解坑了还以为很难。

最长的边的长度小于其余两边之和是组成三角形的充要条件，fft搞搞就行了。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<complex>
using namespace std;
#define LL long long
const int maxn=530010;
double Pi;
typedef complex< double >cd;
cd b[maxn]={};
LL a[maxn]={},cnt[maxn]={};
int bel[maxn]={},s,bt;
void getit(){for(int i=0;i<s;++i)bel[i]=(bel[i>>1]>>1)|((i&1)<<(bt-1));}
void fft(cd *c,int n,int dft){
    for(int i=0;i<n;++i)if(bel[i]>i)swap(c[i],c[bel[i]]);
    for(int step=1;step<n;step<<=1){
        cd w=cd(cos(Pi/(double)step),sin(Pi/(double)step)*(double)dft);
        for(int j=0;j<n;j+=(step<<1)){
            cd z=cd(1.0,0);
            for(int i=j;i<j+step;++i){
                cd x=c[i],y=c[i+step]*z;
                c[i]=x+y;c[i+step]=x-y;
                z=z*w;
            }
        }
    }
    if(dft==-1)for(int i=0;i<n;++i)c[i]/=n;
}
int main(){
    Pi=acos(-1.0);
    int T;scanf("%d",&T);
    while(T-->0){
        int n;scanf("%d",&n);
        memset(cnt,0,sizeof(cnt));
        for(int i=0;i<n;++i){scanf("%d",&a[i]);cnt[a[i]]+=1;}
        
        sort(a,a+n); int siz=a[n-1]+1; 
        for(int i=0;i<siz;++i)b[i]=cd(cnt[i],0);
        for(int i=siz;i<s;++i)b[i]=cd(0,0);
        
        siz*=2; bt=1; s=2; for(;s<siz;++bt)s<<=1; getit();
        fft(b,s,1);
        for(int i=0;i<s;++i)b[i]=b[i]*b[i];
        fft(b,s,-1);
        for(int i=0;i<=s;++i)cnt[i]=(LL)(b[i].real()+0.5);
        for(int i=0;i<s;++i)b[i]=cd(0,0);
        
        s=a[n-1]*2; 
        for(int i=0;i<n;++i)--cnt[a[i]*2];
        for(int i=1;i<=s;++i)cnt[i]/=2;
        for(int i=1;i<=s;++i)cnt[i]+=cnt[i-1];
        
        LL ans=0;
        for(int i=0;i<n;++i){
            ans+=cnt[s]-cnt[a[i]];
            ans-=(LL)(n-1-i)*i;
            ans-=n-1;
            ans-=(LL)(n-1-i)*(n-i-2)/2;
        }
        LL sum=(LL)n*(n-1)*(n-2)/6;
        printf("%.7f
",(double)(ans)/(double)(sum));
    }
    return 0;
}