LA4329 乒乓比赛（树状数组）(训练指南)

题目链接：http://uva.onlinejudge.org/index.php?option=com_onlinejudge&Itemid=8&page=show_problem&problem=4174

考虑第i个人当裁判的情形。假设从a1到ai-1有ci个人比ai小，那么就有(i-1)-ci个比ai大。di同理。

根据乘法原理和加法原理，i当裁判时有ci(n-i-di)+di(i-1-ci)种比赛。

#include <cstdio>
#include <algorithm>
#include <vector>
using namespace std;
const int maxn=20000+10;
int a[maxn],c[maxn],d[maxn];///c[i]表示左边比a[i]小的人数目，d[i]表示右边比a[i]小的人的数目
struct BIT
{
    int n;
    vector<int>C;
    void resize(int n){ this->n=n;C.resize(n);}
    void clear(){ fill(C.begin(),C.end(),0);}
    int lowbit(int x){ return x&(-x);}
    int sum(int x){
        int ret=0;
        while(x>0){
            ret+=C[x];
            x-=lowbit(x);
        }
        return ret;
    }
    void add(int x,int add){
        while(x<=n)
        {
            C[x]+=add;
            x+=lowbit(x);
        }
    }
} f;
int main(){
    int T;
    scanf("%d",&T);
    while(T--){
        int n,maxa=0;
        scanf("%d",&n);
        for(int i=1; i<=n; i++){
            scanf("%d",&a[i]);
            maxa=max(maxa,a[i]);
        }
        f.resize(maxa);
        f.clear();
        for(int i=1;i<=n;i++){
            f.add(a[i],1);
            c[i]=f.sum(a[i]-1);
        }
        f.clear();
        for(int i=n;i>=1;i--){
            f.add(a[i],1);
            d[i]=f.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-1-c[i]);
        printf("%lld\n",ans);
    }
    return 0;
}