hdu 4906 3-idiots fft

题目链接

n个火柴棍取3个， 问能组成三角形的概率是多少。 kuangbin大神的博客写的很详细了..http://www.cnblogs.com/kuangbin/archive/2013/07/24/3210565.html

注意long long什么的就没问题了。

#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define mem(a) memset(a, 0, sizeof(a))
typedef complex <double> cmx;
const double PI = acos(-1.0);
const int maxn = ;
cmx x[maxn];
int a[maxn/];
ll num[maxn];
void change(cmx x[], int len) {
    int i, j, k;
    for(i = , j = len/; i < len - ; i++) {
        if(i < j)
            swap(x[i], x[j]);
        k = len / ;
        while(j >= k) {
            j -= k;
            k /= ;
        }
        if(j < k)
            j += k;
    }
}
void fft(cmx x[], int len, int on) {
    change(x, len);
    for(int i = ; i <= len; i <<= ) {
        cmx wn(cos(-on *  * PI/i), sin(-on *  * PI/i));
        for(int j = ; j < len; j += i) {
            cmx w(, );
            for(int k = j; k < j + i/; k++) {
                cmx u = x[k];
                cmx v = x[k + i/]*w;
                x[k] = u + v;
                x[k+i/] = u - v;
                w *= wn;
            }
        }
    }
    if(on == -) {
        for(int i = ; i < len; i++)
            x[i] /= len;
    }
}
int main()
{
    int t, n;
    cin>>t;
    while (t--) {
        cin>>n;
        mem(num);
        int maxx = ;
        for (int i = ; i < n; i++) {
            scanf("%d", a + i);
            num[a[i]]++;
            maxx = max(maxx, a[i]);
        }
        sort(a, a + n);
        int len = ;
        maxx++;
        while (len < *maxx) {
            len <<= ;
        }
        for (int i = ; i < maxx; i++) {
            x[i] = cmx(num[i], );
        }
        for (int i = maxx; i < len; i++) {
            x[i] = cmx(, );
        }
        fft(x, len, );
        for (int i = ; i < len; i++) {
            x[i] *= x[i];
        }
        fft(x, len, -);
        for (int i = ; i < len; i++) {
            num[i] = (ll)(x[i].real()+0.5);
        }
        for (int i = ; i < n; i++) {
            num[a[i]+a[i]]--;
        }
        for (int i = ; i < len; i++) {
            num[i] /= ;
        }
        for (int i = ; i < len; i++) {
            num[i] += num[i-];
        }
        ll ans = ;
        for (int i = ; i < n; i++) {
            ans += num[len-] - num[a[i]];
            ans -= 1LL * (n-i-) * i;
            ans -= 1LL * (n-i-) * (n-i-) / ;
        }
        ans -= 1LL * n * (n-);
        ll sum = 1LL * n * (n-) * (n-) / ;
        printf("%.7f\n", 1.0*ans/sum);
    }
    return ;
}