题目链接
http://acm.hdu.edu.cn/showproblem.php?pid=4609
题目大意
给出 n <script id="MathJax-Element-2" type="math/tex">n</script>根木棍以及它们的长度,求取其中三根长度各不相同的木棍组成三角形的方案数。
思路
http://blog.csdn.net/qpswwww/article/details/44240723
代码
#include <iostream>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <algorithm>
#include <complex>
#define MAXN 610000
#define PI 3.1415926535897384626
using namespace std;
typedef complex<long double> Complex;
typedef long long int LL;
inline void rev(Complex a[],int n)
{
for(int i=1,j=n/2,k;i<n-1;i++)
{
if(i<j) swap(a[i],a[j]);
k=n/2;
while(j>=k)
{
j-=k;
k>>=1;
}
if(j<k) j+=k;
}
}
inline void FFT(Complex a[],int n,int flag)
{
rev(a,n);
for(int i=1;i<n;i<<=1)
{
Complex wn=Complex(cos(PI/i),flag*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=w*a[j+k+i];
a[j+k]=x+y,a[j+k+i]=x-y;
}
}
}
if(flag==-1)
for(int i=0;i<n;i++)
a[i]=Complex(a[i].real()/n,a[i].imag());
}
Complex a[MAXN];
LL sum[MAXN],len[MAXN],num[MAXN];
int main()
{
int T;
scanf("%d",&T);
while(T--)
{
for(int i=0;i<MAXN;i++) a[i]=Complex(0,0);
LL n=0,N=1,k=0;
scanf("%lld",&n);
for(int i=0;i<n;i++)
{
scanf("%lld",&len[i]);
a[len[i]]=Complex(a[len[i]].real()+1,a[len[i]].imag());
}
sort(len,len+n);
LL tmp=len[n-1]+1; //!!!!!
for(;N<2*tmp+2;N<<=1,k++);
FFT(a,N,1);
for(int i=0;i<N;i++) a[i]=a[i]*a[i];
FFT(a,N,-1);
N=len[n-1]*2; //N=最高次项系数
for(int i=0;i<=N;i++) num[i]=(LL)(a[i].real()+0.5);
for(int i=0;i<n;i++) num[len[i]+len[i]]--; //去掉两个木棍相同的组合
for(int i=0;i<=N;i++) num[i]/=2; //有序变无序
for(int i=1;i<=N;i++) sum[i]=sum[i-1]+num[i];
LL ans=0;
for(int i=0;i<n;i++) //枚举除掉组合的两根木棍外剩下的那根木棍,并假定这根被枚举的木棍是最短的
{
ans+=sum[N]-sum[len[i]]; //ans+=两根长度之和大于本棍子的棍子组合
ans-=(LL)(n-1-i)*i; //减去有一根木棍比本棍短,另一根比本棍长的组合
ans-=n-1; //减去其中一根木棍与本木棍一样长的组合
ans-=(LL)(n-1-i)*(n-2-i)/2; //减去两根木棍都比本木棍长的组合
}
LL all=(LL)n*(n-1)*(n-2)/6; //总组合数
printf("%.7lf\n",(double)ans/all);
}
return 0;
}