可持久化线段树30分求调
查看原帖
可持久化线段树30分求调
782904
哈哈人生楼主2024/4/4 10:15

求调玄关。

评测记录。

MLE可以理解,WA实在不理解。

大概率是query函数的问题?

#include<bits/stdc++.h>
#define int long long
using namespace std;
const int MAXN=100005;
struct TR {
	int v,ls,rs;
} t[MAXN<<8];
int rt[MAXN<<2],cnt=0,ans=0;
void change(int lsto,int &o,int l,int r,int q,int v) {
	if(!o)o=++cnt;
	if(l==r) {
		t[o].v+=v;
		return;
	}
	int mid=l+r>>1;
	if(q<=mid) {
		t[o].rs=t[lsto].rs;
		t[o].ls=++cnt;
		t[t[o].ls]=t[t[lsto].ls];
		change(t[lsto].ls,t[o].ls,l,mid,q,v);
	} else {
		t[o].ls=t[lsto].ls;
		t[o].rs=++cnt;
		t[t[o].rs]=t[t[lsto].rs];
		change(t[lsto].rs,t[o].rs,mid+1,r,q,v);
	}
	t[o].v=t[t[o].ls].v+t[t[o].rs].v;
}
int query(int o1,int o2,int l,int r,int q) {
	if(l==r)return t[o2].v-t[o1].v;
	int mid=l+r>>1;
	if(q<=mid)return query(t[o1].ls,t[o2].ls,l,mid,q);
	else return query(t[o1].rs,t[o2].rs,mid+1,r,q);
}
int T,n,m,a[MAXN<<2],lsh[MAXN<<2],tot;
signed main() {
	ios::sync_with_stdio(false);
	cin.tie(0),cout.tie(0);
	cin>>T;
	while(T--) {
		cin>>n;
		for(int i=1; i<=n; i++)cin>>a[i],lsh[i]=a[i];
		sort(lsh+1,lsh+n+1);
		tot=unique(lsh+1,lsh+n+1)-lsh-1;
		for(int i=1; i<=n; i++) {
			a[i]=lower_bound(lsh+1,lsh+tot+1,a[i])-lsh;
			change(rt[i-1],rt[i],1,tot,a[i],1);
		}
		cin>>m;
		for(int i=1; i<=m; i++) {
			int l,r,q;
			cin>>l>>r>>q;
			cout<<query(rt[l-1],rt[r],1,tot,q)<<"\n";
		}
	}
	return 0;
}
2024/4/4 10:15
加载中...