传送门
题解:按DFS序下标建主席树,然后就不用说什么了吧。。。
注意:如果询问/插入的[l,r]是[0,1e9]的话,就需要多开数组(MAXN*32),而且效率不太高,所以建议先离散化,这样就可以在[0,MAXN]的范围内询问/插入,而且时间可以缩短10%左右。
#include
using namespace std;
const int MAXN=1e5+4;
int n,q,a[MAXN],b[MAXN],in[MAXN],out[MAXN],tot=0;
int root[MAXN]={0},lc[MAXN*18],rc[MAXN*18],sum[MAXN*18],rk[MAXN*18],tim=0;
int head[MAXN],edge=0;
struct EDGE {
int v,nxt;
}e[MAXN<<1];
inline int read() {
int x=0,f=1;char c=getchar();
while (c<'0'||c>'9') {if (c=='-') f=-1;c=getchar();}
while (c>='0'&&c<='9') x=x*10+c-'0',c=getchar();
return x;
}
inline void adde(int u,int v) {
e[edge].nxt=head[u],e[edge].v=v,head[u]=edge++;
e[edge].nxt=head[v],e[edge].v=u,head[v]=edge++;
}
void insert(int pre,int &rt,int l,int r,int val,int id) {
rt=++tim,sum[rt]=sum[pre]+1;
if (l==r) {
rk[rt]=id;
return ;
}
lc[rt]=lc[pre],rc[rt]=rc[pre];
int mid=(l+r)>>1;
if (val<=mid) insert(lc[pre],lc[rt],l,mid,val,id);
else insert(rc[pre],rc[rt],mid+1,r,val,id);
}
int query(int pre,int now,int l,int r,int k) {
if (l==r) return rk[now];
int mid=(l+r)>>1,temp=sum[lc[now]]-sum[lc[pre]];
if (k<=temp) return query(lc[pre],lc[now],l,mid,k);
else return query(rc[pre],rc[now],mid+1,r,k-temp);
}
void dfs(int p,int fa) {
in[p]=++tot;
insert(root[in[p]-1],root[in[p]],0,MAXN,a[p],p);
for (int i=head[p];~i;i=e[i].nxt) {
int v=e[i].v;
if (v^fa)
dfs(v,p);
}
out[p]=tot;
}
int main() {
// freopen("bzoj 1803.in","r",stdin);
memset(head,-1,sizeof(head));
n=read();
for (register int i=1;i<=n;++i) b[i]=a[i]=read();
sort(b+1,b+n+1);
int m=unique(b+1,b+n+1)-b-1;
for (register int i=1;i<=n;++i) a[i]=lower_bound(b+1,b+m+1,a[i])-b;
for (register int i=1;iint u=read(),v=read();
adde(u,v);
}
dfs(1,0);
q=read();
for (register int r=0;r<q;++r) {
int pos=read(),k=read();
printf("%d\n",query(root[in[pos]-1],root[out[pos]],0,MAXN,k));
}
return 0;
}