POJ 3928 Ping Pong

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <ctime>
using namespace std;
int t,n,maxnum;
const int maxm=100000+100;
const int maxn=20000+10;
int g[maxm],h[maxm];
int a[maxn],c[maxn],d[maxn];
int lowbit(int x)
{
return x&(-x);
}
void add(int x,int f[])//是不是一定要是全局数组
{
while(x<=maxnum)//也可以是maxm,不过那样的话好像更费时间,有些用不到的地方也加了1
{
f[x]++;
x+=lowbit(x);
}
}
//这样初始化有问题,f虽然是地址,但是却是指针，size只有4字节
//void init(int *f)
//{
//    printf("in init:%d\n",f);
//    memset(f,0,sizeof(f));
//    printf("fsize:%d\n",sizeof(f));
//    for(int i=0;i<n;i++)
//        printf("%d",f[i]);
//     printf("\n");
//}
int sum(int x,int f[])
{
int ret=0;
while(x>0)
{
ret+=f[x];
x-=lowbit(x);
}
return ret;
}
int main()
{
//  生成随机数测试
//    freopen("test.txt","w",stdout);
//    srand((unsigned)time(NULL));
//    t=rand()%10+1;
//
//    printf("%d\n\n",t);
//    while(t--)
//    {
//        n=rand()%20+1;
//        printf("%d\n",n);
//        for(int i=1;i<=n;i++)
//        {
//            int number=rand()%1000+1;
//            printf("%d ",number);
//        }
//    printf("\n\n");
//    }
//    return 0;
//    freopen("test.txt","r",stdin);
//     freopen("out.txt","w",stdout);
scanf("%d",&t);
while(t--)
{
//        printf("in main:%d\n",a);
scanf("%d",&n);
//        init(a),init(c),init(d),init(g),init(h);这样初始化不对
memset(a,0,sizeof(a));
memset(c,0,sizeof(c));
memset(d,0,sizeof(d));
memset(g,0,sizeof(g));
memset(h,0,sizeof(h));
maxnum=0;
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
maxnum=max(maxnum,a[i]);
}
add(a[1],g);//把树状数组中包含a[1]的长条的值+1
for(int i=2;i<=n;i++)
{
c[i]=sum(a[i],g);//第i个数的左边有c[i]个比a[i]小
add(a[i],g);//把树状数组中包含a[i]的长条的值+1
}
add(a
,h);
for(int i=n-1;i>=1;i--)
{
d[i]=sum(a[i],h);//第i个数的右边有d[i]个比a[i]小
add(a[i],h);
}
long long ans=0;
for(int i=1;i<=n;i++)
ans+=(long long)c[i]*(n-i-d[i])+(long long)d[i]*(i-1-c[i]);
printf("%lld\n",ans);
}
return 0;
}