UVALive 4329 Ping pong

#include <stdio.h>
#include <string.h>

using namespace std;

struct fwtree
{
int C[101000], n;
void init(int k){ memset(C, 0, sizeof(C)); n = k; }
int lowbit(int x) { return x&-x; }
void add(int x, int d)
{
while(x<=n){
C[x] += d;
x += lowbit(x);
}
}
long long sum(int x)
{
long long ans=0;
while(x>=1){
ans += C[x];
x -= lowbit(x);
}
return ans;
}
};

fwtree f;
int main()
{
int t, n;
int maxa;
long long a[20200][4];
int p[20200];
long long ans;

scanf("%d", &t);
while(t--){
scanf("%d", &n);
maxa = -1;
for(int i=1; i<=n; i++){
scanf("%d", &p[i]);
if(p[i]>maxa)   maxa = p[i];
}
f.init(maxa);
ans = 0;
for(int i=1; i<=n; i++){
f.add(p[i], 1);
a[i][0] = f.sum(p[i]-1);        //a[i][0]在左边比a[i]小   a[i][1]为在左边比a[i]大
a[i][1] = i-a[i][0]-1;          //家在i前面的有i-1个
}
f.init(maxa);

for(int i=n; i>=1; i--){
f.add(p[i], 1);
a[i][2] = f.sum(p[i]-1);        //a[i][2]在右边比a[i]小   a[i][3]为在右边比a[i]大
a[i][3] = n-i-a[i][2];
}
for(int i=0; i<n; i++)
ans += a[i][0]*a[i][3] + a[i][1]*a[i][2];
printf("%lld\n", ans);
}

return 0;
}