dalao帮忙改一下,真的改不动了
查看原帖
dalao帮忙改一下,真的改不动了
817681
HZHDCM楼主2024/7/11 20:25

第二个点总是MLE,我又不会别的算法。求dalao调调orz

#include<bits/stdc++.h>
#define int long long
#define lowbit(x) x&(-x)
using namespace std;
const int N=2e5+5;
struct Segment_tree{
	int l,r;
	int sum;
}t[N<<5];
struct node{
	int a,b,c;
}ask[N];
int n,m,a[N],Rank[N],len,cnt,pos,nope,dis[N],tot[N],root[N];
char str[N];
void update(int &rt,int last,int l,int r,int x,int k){
	if(!rt)rt=++cnt;
	t[rt]=t[last];
	if(l==r){
		t[rt].sum+=k;
		return ;
	}
	int mid=(l+r)>>1;
	if(x<=mid)update(t[rt].l,t[last].l,l,mid,x,k);
	else update(t[rt].r,t[last].r,mid+1,r,x,k);
	t[rt].sum=t[t[rt].l].sum+t[t[rt].r].sum;
}
void add(int l,int k){
	for(int i=l;i<=n;i+=lowbit(i)){
		update(root[i],root[i],1,len,a[l],k);
	}
}
int Query(int l,int r,int k){
	while(1){
		int mid=(l+r)>>1;
		if(k<=mid){
			for(int i=1;i<=pos;i++)dis[i]=t[dis[i]].l;
			for(int i=1;i<=nope;i++)tot[i]=t[tot[i]].l;
			r=mid;
		}
		else {
			for(int i=1;i<=pos;i++)dis[i]=t[dis[i]].r;
			for(int i=1;i<=nope;i++)tot[i]=t[tot[i]].r;
			l=mid+1;
		}
		if(l==r)break;
	}
	if(l==r){
		int sum=0;
		for(int i=1;i<=pos;i++)sum+=t[dis[i]].sum;
		for(int i=1;i<=nope;i++)sum-=t[tot[i]].sum;
		return sum;
	}
}
int Find(int l,int r,int k){
	pos=nope=0;
	for(int i=r;i>0;i-=lowbit(i))dis[++pos]=root[i];
	for(int i=l-1;i>0;i-=lowbit(i))tot[++nope]=root[i];
	return Query(1,len,k);
}
signed main(){
	ios::sync_with_stdio(0);
	cin.tie(0);cout.tie(0);
	cin>>n>>m;
	for(int i=1;i<=n;i++)cin>>a[i],Rank[i]=a[i];
	len=n;
	for(int i=1;i<=m;i++){
		cin>>str[i]>>ask[i].a>>ask[i].b;
		if(str[i]=='C')Rank[++len]=ask[i].b;
		else cin>>ask[i].c;
	}
	sort(Rank+1,Rank+len+1);
	len=unique(Rank+1,Rank+len+1)-Rank-1;
	for(int i=1;i<=n;i++){
		a[i]=lower_bound(Rank+1,Rank+len+1,a[i])-Rank;
		add(i,1);
	}
	for(int i=1;i<=m;i++){
		if(str[i]=='C'){
			add(ask[i].a,-1);
			a[ask[i].a]=lower_bound(Rank+1,Rank+len+1,ask[i].b)-Rank;
			add(ask[i].a,1);
		}
		else{
			ask[i].c=lower_bound(Rank+1,Rank+len+1,ask[i].c)-Rank;
			cout<<Find(ask[i].a,ask[i].b,ask[i].c)<<endl;
		}
	}
	return 0;
}
2024/7/11 20:25
加载中...