主席树一跑就RE了,求助,关注奉上
查看原帖
主席树一跑就RE了,求助,关注奉上
524801
不食嗟来之食楼主2022/10/28 17:02
#include<iostream>
#include<vector>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
using namespace std;
const int N=1e5+5;
vector<int> v,e[N];
struct peo{
	int lc,rc;
	int sum;
}a[N<<5];
int getnum(int x){
	return lower_bound(v.begin(),v.end(),x)-v.begin()+1;
}
int n,m,las;
int b[N];
int fa[N][40],t;
int c[N];
int dep[N];
void dfs(int u,int f){
	dep[u]=dep[f]+1;
	fa[u][0]=f;
	for(int i=0;i<=19;i++){
		fa[u][i+1]=fa[fa[u][i]][i];
	}
	for(auto v:e[u]){
		if(v==f)continue;
		dfs(v,u);
	}
}
int lca(int x,int y){
	if(dep[x]<dep[y]) swap(x,y);
	for(int i=19;i>=0;i--){
		if(dep[fa[x][i]]>=dep[y]){
			x=fa[x][i];
		}
		if(x==y) return x;
	}
	for(int i=19;i>=0;i--){
		if(fa[x][i]!=fa[y][i]){
			x=fa[x][i];
			y=fa[y][i];
		}
	}
	return fa[x][0];
}
int rt[N],cnt;
int que(int l,int r,int x,int y,int z,int d,int k){
	if(l==r) return l;
	int res=a[a[x].lc].sum+a[a[y].lc].sum-a[a[z].lc].sum-a[a[d].lc].sum;
	int mid=(l+r)>>1;
	if(k<=res){
		return que(l,mid,a[x].lc,a[y].lc,a[z].lc,a[d].lc,k);
	}
	else return que(mid+1,r,a[x].rc,a[y].rc,a[z].rc,a[d].rc,k-res);
}
void update(int l,int r,int pre,int &now,int val){
	now=++cnt;
	a[now]=a[pre];
	a[now].sum++;
	if(l==r) return ;
	int mid=(l+r)>>1;
	if(val<=mid){
		update(l,mid,a[pre].lc,a[now].lc,val);
	}
	else update(mid+1,r,a[pre].rc,a[now].rc,val);
	return ;
}
void dfs2(int u){
	update(1,n,rt[fa[u][0]],rt[u],c[u]);
	for(auto v:e[u]){
		if(v==fa[u][0]) continue;
		dfs2(v);
	}
}
int main(){
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++){
		scanf("%d",&b[i]);
		v.push_back(b[i]);
	}
	for(int i=1;i<=n-1;i++){
		int u,v;
		scanf("%d%d",&u,&v);
		e[u].push_back(v);
		e[v].push_back(u);
	}
	sort(v.begin(),v.end());
	v.erase(unique(v.begin(),v.end()),v.end()); 
	for(int i=1;i<=n;i++) c[i]=getnum(b[i]);
	t=(int)log(n)/log(2)+1;
	dfs(1,0);
	int ans=0;
	dfs2(1);
	for(int i=1;i<=m;i++){
		int u,vs,k;
		scanf("%d%d%d",&u,&vs,&k);
		u^=ans;
		int z=lca(u,vs);
		las=que(1,n,rt[u],rt[vs],rt[z],rt[fa[z][0]],k);
		ans=v[las];
		printf("%d\n",ans);
	}
	return 0;
}
2022/10/28 17:02
加载中...