UVA 1428 - Ping pong(树状数组)

题目大意：

一条大街上住着n个乒乓球爱好者，经常比赛切磋技术。每个人都有一个不同的技能值a[i]；每场比赛需要3个人：两名选手，一名裁判。他们有个奇怪的约定，裁判必须住在两名选手之间，而裁判的能力值也必须在两名选手之间。问一共能组织多少种比赛。

分析：

假设a[1]到a[i-1]中小于a[i]的数有p[i]，a[i+1]到a[n]中小于a[i]的数有s[i]个;

这样当i为裁判时能够组织的比赛数目为：p[i]*(n-i-s[i]) + (i-1-p[i])*s[i];

AC源码：

#include <iostream>
#include <cstring>
#include <cstdio>
#define LOCA
using namespace std;
typedef long long LL;

const int maxn=1e5+10;
const int N=2e4+10;
LL L[maxn],R[maxn];
LL C[maxn],A[maxn];
LL T,n;
int sum(int i)
{
	LL ret=0;
	while(i>0){
		ret+=C[i];i-=i&(-i);
	}
	return ret;
}
void add(int i,int d)
{
	while(i<maxn){
		C[i]+=d;i+=i&(-i);
	}
}
int main()
{
#ifdef LOCAL
	freopen("data.in","r",stdin);
	freopen("data.out","w",stdout);
#endif
	cin.sync_with_stdio(false);
	cin>>T;
	while(T--){
		memset(C,0,sizeof(C));
		cin>>n;
		for(int i=1;i<=n;++i)
			cin>>A[i];
		for(int i=1;i<=n;++i){
			add(A[i],1);
			L[i]=sum(A[i]-1);
		}
		memset(C,0,sizeof(C));
		for(int i=n;i>=1;--i){
			add(A[i],1);
			R[i]=sum(A[i]-1);
		}
		long long ans=0;
		for(int i=1;i<=n;++i)
			ans+=L[i]*(n-i-R[i])+(i-1-L[i])*R[i];
		cout<<ans<<endl;
	}
	return 0;
}