hdu 2492 Ping pong[树状数组]

题目链接：hdu 2492 Ping pong

题意：(1 <= ai <= 100000，3<=N<=20000)

n个人每个人有个技术值ai，只要三个人中间（位置中间）那个人的技术值在其他两个人中间，他就能成为裁判，并可举办一场比赛，求能举办多少次比赛。

题解：枚举每个人做裁判，用树状数组存每个技术值出现的人数，计算位置在裁判前，比裁判技术值小的人数b[i] 和 位置在裁判后面，比裁判技术值小的人数c[i]，第i个人做裁判能举办的比赛场数即为b[i] * (n-i–c[i]) + c[i] * (i-1-b[i])

#include <iostream>
#include <cstring>
#include <cstdio>
#include <cmath>
#include <algorithm>
using namespace std;
typedef long long ll;
const int N = 2e4+5;
int tree[100005];
int a[N];
int b[N], c[N];//第i人前比他分小的数量，，第i人后比他分小数量
int n, m;
int lowbit(int x){
    return x&(-x);
}
void add(int i){
    while(i <= m){
        tree[i]++;
        i += lowbit(i);
    }
}
int query_sum(int i){
    int sum = 0;
    while(i){
        sum += tree[i];
        i -= lowbit(i);
    }
    return sum;
}
int main() {
    int i, t;
    scanf("%d", &t);
    while(t--) {
        memset(tree, 0, sizeof(tree));
        scanf("%d", &n);
        m = 0;
        for(i = 1; i <= n; ++i) {
            scanf("%d", &a[i]);
            m = max(m, a[i]);
        }
        for(i = 1; i <= n; ++i) {
            add(a[i]);
            b[i] = query_sum(a[i]-1);
        }
        memset(tree, 0, sizeof(tree));
        for(i = n; i >= 1; --i) {
            add(a[i]);
            c[i] = query_sum(a[i]-1);
        }
        ll ans = 0;
        for(i = 2; i <= n-1; ++i)
            ans += b[i] * (n-i-c[i]) + c[i] * (i-b[i]-1);
        printf("%lld\n", ans);
    }
    return 0;
}