uva 1428 - Ping pong （树状数组的应用）

题意：n个人，每个人有个居住编号和技能度，选3个人（其中1个是裁判）来比赛，要求裁判居住在两选手之间，技能值也要在两选手之间，一共有多少种比赛？

解答： i 为裁判，则c【i】为i前面技能比他小的人数，d【i】为后面技能比他小的人数，则答案为

c[i]*(n-d[i]-i)+d[i]*(i-c[i]-1);

#include<iostream>
#include<cstdio>
#include<string.h>
#include<algorithm>
using namespace std;
typedef long long LL;
#define N 20005
#define max 100005
LL a[N],c[N],d[N],x[max];

int lowbit(int i)
{
    return i&-i;
}
void add(int i,int k)
{
    while(i<max)
    {
        x[i]+=k;
        i=i+lowbit(i);
    }
}

LL sum(int n)
{
    LL res=0;
    while(n>0)
    {
        res+=x[n];
        n=n-lowbit(n);
    }
    return res;
}
int main()
{
    int i,j,n,t;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d",&n);
        memset(c,0,sizeof(c));
        memset(d,0,sizeof(d));
        memset(x,0,sizeof(x));
        for(i=1; i<=n; i++)
        {
            scanf("%d",&a[i]);
            add(a[i],1);
            c[i]=sum(a[i]-1);
        }
        for(i=1; i<=n; i++)
        {
            d[i]=sum(a[i]-1)-c[i];
        }
        LL res=0;
        for(i=2; i<n; i++)
        {
            res=res+c[i]*(n-d[i]-i)+d[i]*(i-c[i]-1);
        }
        printf("%lld\n",res);
    }
    return 0;
}