題意：

給定n個數(shù)，隨機(jī)從這n個數(shù)中取3個數(shù)，問能組成三角形的概率是多少？

思路：

首先把統(tǒng)計這n個數(shù)出現(xiàn)的個數(shù)，那么會得到一個向量，這個向量的自我的乘積就是a[i]+a[j]的可能的方案數(shù)，這樣，我么就很方便求出了兩條邊的和的方案數(shù)。

但是在加的過程中，我么多加了a[i]+a[i]的情況，也就是自己跟自己相加的情況，所以我們要減去這一段。而且，在加的過程a[i]+a[j]和a[j]+a[i]是同一種情況，我們加了兩次，所以要減去一半。

sum是num的前綴和，假設(shè)a[i]是三角形內(nèi)要取得最長的邊，那么只需要在前面取兩條邊并且這兩條邊的和大于a[i];
而長度和大于a[i]的取兩個的取法是sum[len]-sum[a[i]].

但是這里面有不符合的。
一個是包含了取一大一小的
cnt -= (long long)(n-1-i)*i;
一個是包含了取一個本身i,然后取其它的
cnt -= (n-1);
還有就是取兩個都大于的了
cnt -= (long long)(n-1-i)*(n-i-2)/2;
這樣把i從0~n-1累加，就答案了。

參考了kuangbin 的博客：
http://www.cnblogs.com/kuangbin/archive/2013/07/24/3210565.html

code：

#include <iostream> #include <cstdio> #include <cstring> #include <cmath> #include <algorithm> using namespace std;const double PI=acos(-1.0); typedef long long ll;struct complex {double l,r;complex(double ll=0.0,double rr=0.0){l=ll;r=rr;}complex operator +(const complex& B){return complex(l+B.l,r+B.r);}complex operator - (const complex& B){return complex(l-B.l,r-B.r);}complex operator *(const complex& B){return complex(l*B.l-r*B.r,l*B.r+B.l*r);} };/** 進(jìn)行FFT和IFFT前的反轉(zhuǎn)變換。* 位置i和j（i二進(jìn)制反轉(zhuǎn)后位置）互換* len必須是2的冪*/ void change(complex y[],int len){int i,j,k;for (int 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>>=1;}if (j<k) j+=k;} } /** 做FFT* len必須為2^k形式，* on==1時是DFT，on==-1時是IDFT*/ 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].l/=len;}} } const int MAXN=400040; complex x1[MAXN]; ll sum[MAXN],num[MAXN]; int a[MAXN/4];int main() {int T,n;scanf("%d",&T);while (T--){scanf("%d",&n);memset (num,0,sizeof(num));for (int i=0;i<n;i++){scanf("%d",a+i);num[a[i]]++;}sort(a,a+n);int len1=a[n-1]+1,len=1;while (len<2*len1) len<<=1;for (int i=0;i<len1;i++){x1[i]=complex(num[i],0);}for (int i=len1;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].l+0.5);}len=2*a[n-1];for (int i=0;i<n;i++) num[a[i]+a[i]]--;for (int i=1;i<=len;i++) num[i]/=2;sum[0]=0;for (int i=1;i<=len;i++) sum[i]=sum[i-1]+num[i];ll cnt=0;for (int i=0;i<n;i++){cnt+=sum[len]-sum[a[i]];cnt-=(ll)(n-i-1)*i;cnt-=(n-1);cnt-=(ll) (n-1-i)*(n-i-2)/2;}ll tot=(ll)n*(n-1)*(n-2)/6;printf("%.7f\n",1.0*cnt/tot);} }

總結(jié)

以上是生活随笔為你收集整理的HDU 4609——3-idiots的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。