权值线段树 TLE on 5~12 求掉

#include<bits/stdc++.h>
#define int long long
#define lc f[p].l
#define rc f[p].r
using namespace std;
const int N=5e6+5;
int n,m,k;
int T;
struct node{
	int l,r,ans,mx,mn;
}f[N];
int idx;
void update(int &p,int L,int R,int x,int k){
	if(!p){
		p=++idx;
		f[p].mx=-1e7;
		f[p].mn=1e7;
	}	
	if(R<x||L>x)return;
	if(L==R){
		f[p].ans+=k;
		if(f[p].ans>0){
			f[p].mx=L,f[p].mn=L;
		}
		else{
			f[p].mx=-1e7,f[p].mn=1e7;
		}
		return;
	}
	int mid=L+R>>1;
	update(lc,L,mid,x,k);
	update(rc,mid+1,R,x,k);
	f[p].ans=f[lc].ans+f[rc].ans;
	f[p].mx=max(f[lc].mx,f[rc].mx);
	f[p].mn=min(f[lc].mn,f[rc].mn);
}
int query(int p,int L,int R,int l,int r){
	if(!p)return 0;
	if(l<=L&&R<=r){
		return f[p].ans;
	}
	int mid=L+R>>1;
	int res=0;
	if(l<=mid)res+=query(lc,L,mid,l,r);
	if(r>mid)res+=query(rc,mid+1,R,l,r);
	return res;
}
int querymx(int p,int L,int R,int l,int r){
	if(!p)return -1e7;
	if(l<=L&&R<=r){
		return f[p].mx;
	}
	int mid=L+R>>1;
	int res=-1e7;
	if(l<=mid)res=max(res,querymx(lc,L,mid,l,r));
	if(r>mid)res=max(res,querymx(rc,mid+1,R,l,r));
	return res;
}
int querymn(int p,int L,int R,int l,int r){
	if(!p)return 1e7;
	if(l<=L&&R<=r){
		return f[p].mn;
	}
	int mid=L+R>>1;
	int res=1e7;
	if(l<=mid)res=min(res,querymn(lc,L,mid,l,r));
	if(r>mid)res=min(res,querymn(rc,mid+1,R,l,r));
	return res;
}
int rt;
int check(int mid,int x){
	int mn=query(rt,-1e7,1e7,-1e7,mid-1);
	int num=query(rt,-1e7,1e7,mid,mid);
	if(mn+1<=x&&mn+num>=x){
		return 1;
	}
	if(x<mn+1)return 2;
	return 3;
}
signed main(){
//	ios::sync_with_stdio(0);
	cin>>m;
	while(m--){
		int op;
		cin>>op;
		if(op==1){
			int x;
			cin>>x;
			update(rt,-1e7,1e7,x,1);
		}
		else if(op==2){
			int x;
			cin>>x;
			update(rt,-1e7,1e7,x,-1);
		}
		else if(op==3){
			int x;
			cin>>x;
			cout<<query(rt,-1e7,1e7,-1e7,x-1)+1<<'\n';
		}
		else if(op==4){
			int x;
			cin>>x;
			int l=-1e7,r=1e7;
			while(l<r){
				int mid=l+1+r>>1;
				int ch=check(mid,x);
				if(ch==1)r=mid;
				else if(ch==2)r=mid-1;
				else l=mid+1;
			}
			if(!query(rt,-1e7,1e7,r,r))cout<<querymn(rt,-1e7,1e7,r+1,1e7)<<'\n';
			else cout<<r<<'\n';
		}
		else if(op==5){
			int x;
			cin>>x;
			int Pre=querymx(rt,-1e7,1e7,-1e7,x-1);
			cout<<Pre<<'\n';
		}
		else{
			int x;
			cin>>x;
			int Nex=querymn(rt,-1e7,1e7,x+1,1e7);
			cout<<Nex<<'\n';
		}
	}
}