hdu2492 Ping pong (树状数组)

#include <stdio.h>
#include <string.h>
#define MAX 100005

int arr[MAX],a[MAX],n;
int leftMin[MAX],leftMax[MAX],rightMin[MAX],rightMax[MAX];

int lowBit(int x)
{
	return (x&(-x));
}

void add(int index,int val)
{
	while(index<=MAX)
	{
		arr[index]+=val;
		index+=lowBit(index);
	}
}

int func(int num)
{
	int sum=0;
	while(num>0)
	{
		sum+=arr[num];
		num-=lowBit(num);
	}
	return sum;
}
int main()
{
	int t,i;
	_int64 ans;
	scanf("%d",&t);
	while(t--)
	{
		ans=0;
		scanf("%d",&n);
		for (i=1;i<=n;i++)
		{
			scanf("%d",&a[i]);
		}
		memset(arr,0,sizeof(arr));
		for (i=1;i<=n;i++)
		{
			add(a[i],1);
			leftMin[i]=func(a[i]-1);
			leftMax[i]=func(MAX)-func(a[i]);
		}
		memset(arr,0,sizeof(arr));
		for(i=n;i>=1;i--)
		{
			add(a[i],1);
			rightMin[i]=func(a[i]-1);
			rightMax[i]=func(MAX)-func(a[i]);
		}
		for(i=1;i<=n;i++)
			ans+=leftMin[i]*rightMax[i]+leftMax[i]*rightMin[i];
		printf("%I64d\n",ans);
	}
	return 0;
}