LA 4329 Ping pong

题目：Ping pong

思路：树状数组。设a[i]是一个人i的数值。根据乘法原理可以看出，对于每一个人i而言，以他为裁判一共可以举行的比赛场数= 在他左边数值比他小的(c[i]) * 在他右边数值比他大的(n-i-d[i]) + 在他左边数值比他大的 (i-1-c[i]) *在他右边数值比他小的 (d[i]) 。此时，要求c[i]和d[i]，就可以用树状数组。对于每一个裁判i，先设f[j]=1为i左边有一个人的数值为j，c[i] 就等于 f[1] ~ f[i-1]的值。因为有多个裁判，所以f的值是不定的，所以可以将f构造成树状数组求解了。同理可以求出d。

代码：

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <stack>
#include <queue>
#include <deque>
#include <set>
#include <cstring>
#include <map>
using namespace std;

int n;
int a[20005]={0};

int c[20005]={0};
int d[20005]={0};

int f[100005]={0};

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

void add(int x,int d){
	while(x<=100005){	//此处不应为x<=n 
		f[x]+=d;
		x+=lowbit(x);
	}
}

int sum(int x){
	int y=0; 
	while(x>0){
		y+=f[x];
		x-=lowbit(x);
	}
	return y;
}

int main() {
	
	int T;
	scanf("%d",&T);
	while(T--){
		
		memset(a,0,sizeof(a));
		memset(c,0,sizeof(c));
		memset(d,0,sizeof(d));
		memset(f,0,sizeof(f));
		
		scanf("%d",&n);
		for(int i=1;i<=n;i++){
			scanf("%d",&a[i]);
		}
		
		add(a[1],1);
		for(int i=2;i<n;i++){
			c[i]=sum(a[i]-1);	//不计算当前的数，所以-1 
			add(a[i],1);
		}
		memset(f,0,sizeof(f));
		add(a[n],1);
		for(int i=n-1;i>=2;i--){
			d[i]=sum(a[i]-1);
			add(a[i],1);
		}
		
		long long ans=0;
		for(int i=2;i<n;i++){
			ans=ans+c[i]*(n-i-d[i])+(i-1-c[i])*d[i];
		}
		
		printf("%lld\n",ans);
		
	}
	
	return 0;
}