codeforces-61E (线段树求三元逆序对)

题意：给出一个序列，求三元逆序对

题解：枚举中间的数就好了，那么结果就是它左边大于它的数的个数*右边小于它的数的个数，和求线段树二元逆序对的方法是一样的。

线段树求二元逆序对方法：

首先离散化，从左往右扫一遍序列，每一次查询区间（a[i], n）的和，然后更新使a[i]叶子的值自增1。

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

using namespace std;

int n;
int a[1000010];
int b[1000010];

long long lnum[1000010];
long long rnum[1000010];

struct node{
    int l, r;
    long long ma;
}s[1000010*4];

void build(int l, int r, int k){
    s[k].l = l; s[k].r = r;
    if(l == r){
        s[k].ma = 0;
        return;
    }
    int mid = (l+r)>>1;
    build(l, mid, 2*k);
    build(mid+1, r, 2*k+1);
}
long long query(int l, int r, int k){
    if(s[k].l == l && s[k].r == r){
        return s[k].ma;
    }
    int mid = (s[k].l+s[k].r)>>1;
    if(r<=mid) return query(l, r, 2*k);
    else if(l>mid) return query(l, r, 2*k+1);
    else return query(l, mid, 2*k)+query(mid+1, r, 2*k+1);
}

void update(int x, int k){
    if(s[k].l == s[k].r && s[k].l == x){
        s[k].ma++;
        return;
    }
    int mid = (s[k].l+s[k].r)>>1;
    if(x<=mid) update(x, 2*k);
    else update(x, 2*k+1);
    s[k].ma = s[2*k].ma+s[2*k+1].ma;
}


int main(){
    scanf("%d", &n);
    for(int i = 1; i<=n; i++){
        scanf("%d", &a[i]);
        b[i] = a[i];
    }
    sort(b+1, b+1+n);
    int cnt = unique(b+1, b+1+n)-(b+1);
    for(int i = 1; i<=n; i++){
        a[i] = lower_bound(b+1, b+1+cnt, a[i])-b;
    }
    /*for(int i = 1; i<=n; i++){
        cout<<a[i]<<endl;
    }*/
    build(1, n, 1);
    for(int i = 1; i<=n; i++){
        lnum[i] = query(a[i], n, 1);
        update(a[i], 1);
    }
    /*for(int i = 1; i<=n; i++){
        cout<<lnum[i]<<endl;
    }*/

    for(int i = 0; i<=4000010; i++){
        s[i].ma = 0;
    }
    for(int i = n; i>=1; i--){
        rnum[i] = query(1, a[i], 1);
        update(a[i], 1);
    }
    long long res = 0;
    for(int i = 1; i<=n; i++){
       res += lnum[i]*rnum[i];
    }
    cout<<res<<endl;


    return 0;
}