HDU 4609(3-idiots)

题意：给定 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

*/