UVa1428 - Ping pong

我有话说：这道题可以算是树状数组入门经典。要求在两个点之间找一个点，且这个点的值在两个点之间，求有这样的三点集合共有多少个。
这道题我们可以先后找一个点，假设a1到ai-1中有ci个小于ai，ai+1到an中有di个小于ai。那么以该店为中间点则共有ci（n-i-di)+di(i-1-ci)种点集。则转换为求ci和di。
我们这样计算ci，从左到右扫描ai令xi表示目前为止已考虑过的ai中是否存在一个ai=j（x[j]表示这样a[i]的个数）。那么ci就是前缀和x1+x2+……+x[ai-1]。求d类似。

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;
const int maxn=20000+10;
const int maxa=100000+10;
int n,_maxa;
int a[maxn],C[maxa],c[maxn],d[maxn];
int lowbit(int x)
{
    return -x&x;
}
void add(int x,int d)
{
    while(x<=_maxa){
        C[x]+=d;
        x+=lowbit(x);
    }
}
int sum(int x)
{
    int ret=0;
    while(x>0){
      ret+=C[x];
      x-=lowbit(x);
    }
    return ret;
}
int main()
{
    int T;
    cin>>T;
    while(T--){
        scanf("%d",&n);
        _maxa=0;
        for(int i=1;i<=n;i++){
            scanf("%d",&a[i]);
            _maxa=max(_maxa,a[i]);
        }
        memset(C,0,sizeof(C));
        for(int i=1;i<=n;i++){
            add(a[i],1);
            c[i]=sum(a[i]-1);
        }
        memset(C,0,sizeof(C));
        for(int i=n;i>=1;i--){
            add(a[i],1);
            d[i]=sum(a[i]-1);
        }
        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-c[i]-1);
        }
        printf("%lld\n",ans);
    }
    return 0;
}