寒假刷题20：Ping pong （UVALive - 4329）树状数组

题目链接：

Ping Pong

题目解析：

考虑第i个人当裁判，设此时左边又x个人比他小，右边有y个人比他小。那么，他能组织x*(n-i-y)+(i-1-x)*y场比赛把n个人的场次累加即可。

接下来考虑如何求x和y。从左到右扫描数组，设a[j]表示目前为止数字j出现了几次，则x就是。类似的，可以求出y。

这个修改、求和的过程就要用到树状数组。

AC代码（网站好像有问题......最近一页这道题都是WA）：

#include <iostream>
#include <cstring>
#include <cstdio>

using namespace std;

const int maxn = 100000 + 100;
int t, n, a[maxn], c[maxn], b[maxn], d[maxn];

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

void add(int x){
    while(x<=maxn){
        c[x] += 1;
        x += lowbit(x);
    }
}

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

int main(){ 
    cin>>t;
    while(t--){
        cin>>n;
        for(int i=1; i<=n; ++i) cin>>a[i];
        memset(c, 0, sizeof(c));
        for(int i=1; i<=n; ++i){
            add(a[i]);
            b[i] = sum(a[i]-1);
        }
        memset(c, 0, sizeof(c));
        for(int i=n; i>0; --i){
            add(a[i]);
            d[i] = sum(a[i]-1);
        }
        long long ans = 0;
        for(int i=1; i<n; ++i){
            ans += (long long)b[i]*(n-i-d[i]) + (long long)(i-b[i]-1)*d[i]; 
        }
        cout<<ans<<endl;
    }
    return 0;
}