练习33，三元上升子序列【线段树/离散化】

逆序对升级版，在某次校赛遇到过，

P1637 三元上升子序列 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

思路：三元上升子序列的个数=(每个数前比它小的数的个数*每个数后比它大的数的个数) 之和，所以就正求一遍顺序对，反求一遍逆序对就行

#include<bits/stdc++.h>
using namespace std;
#define INF 0x3f3f3f3f
#define NOTLE ios::sync_with_stdio(0),cin.tie(0),cout.tie(0)
#define endl '\n'
#define lint long long
lint a[500010],fa[500010],b[500010],c[500010];
lint tree[2000010]={0},gt[500010]={0},lt[500010]={0};
//gt存每个数后比它大的数的个数
//lt存每个数前比它小的数的个数
lint maxn=0,ans=0,x;
int bs(lint ob,int l,int r){ //二分查找实现离散化
    while(l<r){
        int mid=(l+r+1)/2;
        if(ob>=b[mid]) l=mid;
        else r=mid-1;
    }
    return r;
}

void update(lint i,int l,int r){
    if(l==r){tree[i]++; return ;}
    int mid=(l+r)/2;
    if(x<=mid) update(i*2,l,mid);
    else update(i*2+1,mid+1,r);
    tree[i]=tree[i*2]+tree[i*2+1];
}

lint query1(lint i,int l,int r){ //找顺序对
    if(0<=l && x>=r) return tree[i];
    int mid=(l+r)/2;
    lint add=0;
    if(mid>=x) add+=query1(i*2,l,mid);
    if(mid<x) add+=query1(i*2,l,mid)+query1(i*2+1,mid+1,r);
    return add;
}
lint query2(lint i,int l,int r){ //找逆序对
    if(x<=l && maxn>=r) return tree[i];
    int mid=(l+r)/2;
    lint add=0;
    if(mid>=x) add+=query2(i*2,l,mid);
    if(mid<maxn) add+=query2(i*2+1,mid+1,r);
    return add;
}
int main(){
    NOTLE;
    int n;
    cin >> n;
    for(int i=1;i<=n;i++){
        cin >> a[i];
        fa[i]=a[i];
    }
    sort(fa+1,fa+1+n);

    //离散化，用相对大小关系表示绝对大小
    int cnt=1;
    b[cnt]=fa[1];
    for(int i=2;i<=n;i++)
        if(fa[i]!=fa[i-1]) b[++cnt]=fa[i];
    for(int i=1;i<=n;i++){
        c[i]=bs(a[i],1,cnt+1);
        maxn=max(maxn,c[i]); //找出最大值作为线段树右边
    }

    for(int i=1;i<=n;i++){ //查找每个数前比它小的数的个数
        x=c[i]-1;
        lt[i]=query1(1,0,maxn);
        x=c[i];
        update(1,0,maxn);//将c[i]加入线段树
    }

    memset(tree,0,sizeof(tree)); //清空线段树

    for(int i=n;i>=1;i--){//查找每个数后比它大的数的个数
        x=c[i]+1;
        gt[i]=query2(1,0,maxn);
        x=c[i];
        update(1,0,maxn);//将c[i]加入线段树
    }

    //三元上升子序列的个数=(每个数前比它小的数的个数*每个数后比它大的数的个数)的和
    for(int i=1;i<=n;i++) ans+=gt[i]*lt[i];
    cout << ans;
    return 0;
}