bzoj 1803: Spoj1487 Query on a tree III(主席树)

时间:2022-01-03 19:31:15

题意

你被给定一棵带点权的n个点的有根数,点从1到n编号。

定义查询 query(x,k): 寻找以x为根的k大点的编号(从小到大排序第k个点)

假设没有两个相同的点权。

输入格式: 第一行为整数n,第二行为点权,接下来n-1行为树边,接下来一行为整数m,下面m行为两个整数x,k,代表query(x,k)

输出格式: m行,输出每次查询的结果。

题解

先一遍dfs,然后建个主席树,带上去直接跑一跑就好了

我忘了注意dfs序的位置和原来的编号……结果调了半天啥都调不出来……

 //minamoto
#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
#define getc() (p1==p2&&(p2=(p1=buf)+fread(buf,1,1<<21,stdin),p1==p2)?EOF:*p1++)
char buf[<<],*p1=buf,*p2=buf;
inline int read(){
#define num ch-'0'
char ch;bool flag=;int res;
while(!isdigit(ch=getc()))
(ch=='-')&&(flag=true);
for(res=num;isdigit(ch=getc());res=res*+num);
(flag)&&(res=-res);
#undef num
return res;
}
char obuf[<<],*o=obuf;
inline void print(int x){
if(x>) print(x/);
*o++=x%+;
}
const int N=,M=N*;
int sum[M],L[M],R[M],rt[N];
int ver[N<<],Next[N<<],head[N];
int ls[N],rs[N],a[N],b[N],id[N],pos[N];
int n,m,cnt,tot,q;
void update(int last,int &now,int l,int r,int x){
sum[now=++cnt]=sum[last]+;
if(l==r) return;
int mid=(l+r)>>;
if(x<=mid) R[now]=R[last],update(L[last],L[now],l,mid,x);
else L[now]=L[last],update(R[last],R[now],mid+,r,x);
}
int query(int u,int v,int l,int r,int k){
if(l>=r) return l;
int x=sum[L[v]]-sum[L[u]];
int mid=(l+r)>>;
if(x>=k) return query(L[u],L[v],l,mid,k);
else return query(R[u],R[v],mid+,r,k-x);
}
inline void add(int u,int v){
ver[++tot]=v,Next[tot]=head[u],head[u]=tot;
ver[++tot]=u,Next[tot]=head[v],head[v]=tot;
}
void dfs(int u,int fa){
a[ls[u]=++m]=b[u],id[m]=u;
for(int i=head[u];i;i=Next[i])
if(ver[i]!=fa) dfs(ver[i],u);
rs[u]=m;
}
int main(){
//freopen("testdata.in","r",stdin);
n=read();
for(int i=;i<=n;++i) b[i]=read();
for(int i=;i<n;++i){
int u,v;
u=read(),v=read();
add(u,v);
}
dfs(,);
sort(b+,b++m);
for(int i=;i<=n;++i){
int k=lower_bound(b+,b++m,a[i])-b;
update(rt[i-],rt[i],,m,k);
pos[k]=id[i];
}
q=read();
while(q--){
int u=read(),k=read();
int ans=pos[query(rt[ls[u]-],rt[rs[u]],,m,k)];
print(ans),*o++='\n';
}
fwrite(obuf,o-obuf,,stdout);
return ;
}