HDU 4609 3-idiots

#include<bits/stdc++.h>
using namespace std;
#define fi first
#define se second
#define pi acos(-1.0)
#define LL long long
#define mp make_pair
#define pb push_back
#define ls rt<<1, l, m
#define rs rt<<1|1, m+1, r
#define ULL unsigned LL
#define pll pair<LL, LL>
#define pii pair<int, int>
#define piii pair<int,pii>
#define mem(a, b) memset(a, b, sizeof(a))
#define fio ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#define fopen freopen("in.txt", "r", stdin);freopen("out.txt", "w", stout);
//head

const int N = 4e5 + 10;
int a[N], R[N];
LL num[N],sum[N];
struct Complex {
    double x, y;
    Complex(double _x=0, double _y=0) : x(_x),y(_y) {};
    Complex operator + (Complex &t) {return Complex(x+t.x, y+t.y);}
    Complex operator - (Complex &t) {return Complex(x-t.x, y-t.y);}
    Complex operator * (Complex &t) {return Complex(x*t.x - y*t.y, x*t.y + y*t.x);}
}X[N];
void fft(Complex *x, int n, int type) {
    for (int i = 0; i < n; i++) if(i < R[i]) swap(x[i], x[R[i]]);
    for (int i = 1; i < n; i<<=1) {
        Complex wn(cos(pi/i), type*sin(pi/i));
        for (int j = 0; j < n; j+=i<<1) {
            Complex w(1, 0);
            for (int k = 0; k < i; k++, w=w*wn) {
                Complex X = x[j+k];
                Complex Y = w*x[j+k+i];
                x[j+k] = X + Y;
                x[j+k+i] = X - Y;
            }
        }
    }
}
int main() {
    int T, n;
    scanf("%d", &T);
    while(T--) {
        mem(num, 0);
        scanf("%d", &n);
        for (int i = 0; i < n; i++) scanf("%d", &a[i]), num[a[i]]++;
        sort(a, a+n);
        int len = a[n-1] + 1;
        int l = 1, L = 0;
        for (int i = 0; i < len; i++) X[i] = Complex(num[i], 0);
        for ( ; l < len*2; l <<= 1) L++;
        for (int i = len; i < l; i++) X[i] = Complex(0, 0);
        for (int i = 0; i < l; i++) {
            R[i] = (R[i>>1]>>1)|((i&1)<<L-1);
        }
        fft(X, l, 1);
        for (int i = 0; i < l; i++) X[i] = X[i]*X[i];
        fft(X, l, -1);
        for (int i = 0; i < l; i++) num[i] = (LL)(X[i].x/l + 0.5);
        for (int i = 0; i < n; i++) num[a[i]+a[i]]--; 
        for (int i = 0; i < l; i++) num[i] /= 2;
        sum[0] = num[0];
        for (int i = 1; i < l; i++) sum[i] = sum[i-1] + num[i]; 
        LL ans = 0;
        for(int i = 0; i < n; i++) {
            ans += sum[l-1] - sum[a[i]];
            ans -= 1LL * (n-i-1) * i;
            ans -= 1LL * (n-i-1) * (n-i-2) / 2;
            ans -= n-1;
        } 
        LL t = 1LL*n*(n-1)*(n-2)/6;
        printf("%.7f\n", (double)ans/t);
    }
    return 0;
}