三元上升子序列(线段树)

#pragma GCC optimize(3,"Ofast","inline")
#pragma comment(linker, "/STACK:1024000000,1024000000")
#include<bits/stdc++.h>
using namespace std;
#define endl '\n'
#define int long long
const int N=30010;
int n,a[N],L[N],R[N];
/*
三元上升子序列，或者下降子序列；
取该数字，l[i]表示，左边小于他的个数
        r[i]表示，大于他的个数，
相乘即可，线段树维护区间l-r的值
*/
struct node{
	int l,r,sum;
}tr[N*4];
vector<int>alls;
void pushup(int u){
	tr[u].sum=tr[u*2].sum+tr[u*2+1].sum;
}
void build(int u,int l,int r){
	tr[u]={l,r};
	if(l==r){
		tr[u].sum=0;
		return;
	}
	int mid=(l+r)/2;
	build(u*2,l,mid);
	build(u*2+1,mid+1,r);
	pushup(u);	
}
void change(int u,int x){
	if(tr[u].l==tr[u].r) {
		tr[u].sum++;
		return;
	}
	int mid=(tr[u].l+tr[u].r)/2;
	if(mid>=x){//在左区间
		change(u*2,x);
	}
	else change(u*2+1,x);
	pushup(u);
}
int query(int u,int l,int r){
	if(tr[u].l>=l&&tr[u].r<=r) {
		return tr[u].sum;
	}	
	int sum=0;
	int mid=(tr[u].l+tr[u].r)/2;
	if(l<=mid) sum+=query(u*2,l,r);
	if(r>mid) sum+=query(u*2+1,l,r);
	return sum;
}
void solve()
{
	cin>>n;
	for(int i=1;i<=n;i++){
		cin>>a[i];
		alls.push_back(a[i]);
	}
	sort(alls.begin(),alls.end());//离散化
	alls.erase(unique(alls.begin(),alls.end()),alls.end());
	build(1,1,n);
	for(int i=1;i<=n;i++){//下标查询,+1从1开始
		int id=lower_bound(alls.begin(),alls.end(),a[i])-alls.begin()+1;
		L[i]=query(1,1,id-1);
		change(1,id);
	}
	build(1,1,n);
	for(int i=n;i>=1;i--){
		int id=lower_bound(alls.begin(),alls.end(),a[i])-alls.begin()+1;
		R[i]=query(1,id+1,n);
		change(1,id);
	}
	int ans=0;
	for(int i=1;i<=n;i++) {ans+=L[i]*R[i];}
	cout<<ans;
}
signed main()
{
	ios::sync_with_stdio(false);
cin.tie(0),cout.tie(0);
	solve();
	return 0;
}

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