标签：inpu name std markdown follow query nod 就是 sdi
You are given a node-labeled rooted tree with n nodes.
Define the query (x, k): Find the node whose label is k-th largest in the subtree of the node x. Assume no two nodes have the same labels.
The first line contains one integer n (1 <= n <= 105). The next line contains n integers li (0 <= li <= 109) which denotes the label of the i-th node.
Each line of the following n - 1 lines contains two integers u, v. They denote there is an edge between node u and node v. Node 1 is the root of the tree.
The next line contains one integer m (1 <= m <= 104) which denotes the number of the queries. Each line of the next m contains two integers x, k. (k <= the total node number in the subtree of x)
For each query (x, k), output the index of the node whose label is the k-th largest in the subtree of the node x.
5
1 3 5 2 7
1 2
2 3
1 4
3 5
4
2 3
4 1
3 2
3 2
5
4
5
5
查询第k大就是主席树裸体
然后 这个就是查询子树第k大
那不就是 dfs序上查询即可？
#include <bits/stdc++.h>
using namespace std;
#define maxn (int)(1e5+10)
#define LL long long
inline int read(){
int rtn=0,f=1;char ch=getchar();
while(!isdigit(ch)){if(ch=='-')f=-1;ch=getchar();}
while(isdigit(ch))rtn=(rtn<<1)+(rtn<<3)+ch-'0',ch=getchar();
return rtn*f;
}
int n,m,cnt,p[maxn],w[maxn],dfn[maxn],root[maxn],size[maxn],coc,num,sz;
struct node{
int a,b,nt;
}e[maxn<<2];
struct HJT{
int ls,rs,cnt,id;
}T[maxn*60];
inline void add(int x,int y){
e[++cnt].a=x;e[cnt].b=y;
e[cnt].nt=p[x];p[x]=cnt;
}
inline void insert(int &p,int l,int r,int q,int v){
T[++sz]=T[p];p=sz;
if(l==r)return void((T[p].cnt=1)&&(T[p].id=v));
int mid=l+r>>1;
if(q<=mid)insert(T[p].ls,l,mid,q,v);
else if(q>mid)insert(T[p].rs,mid+1,r,q,v);
T[p].cnt=T[T[p].ls].cnt+T[T[p].rs].cnt;
}
inline int query(int lp,int rp,int l,int r,int k){
if(l==r)return T[rp].id;
int s=T[T[rp].ls].cnt-T[T[lp].ls].cnt;
int mid=l+r>>1;
if(k<=s)return query(T[lp].ls,T[rp].ls,l,mid,k);
else return query(T[lp].rs,T[rp].rs,mid+1,r,k-s);
}
inline void dfs(int x){
size[x]=1;dfn[x]=++coc;
root[coc]=root[coc-1];
insert(root[coc],0,1e9,w[x],x);
for(int i=p[x];i;i=e[i].nt){
int k=e[i].b;
if(size[k])continue;
dfs(k);size[x]+=size[k];
}
}
int main(){
n=read();
for(int i=1;i<=n;i++)scanf("%d",&w[i]);
for(int i=1;i<n;i++){
int x=read(),y=read();
add(x,y);add(y,x);
}
dfs(1);
m=read();
while(m--){
int x=read(),k=read();
printf("%d\n",query(root[dfn[x]-1],root[dfn[x]+size[x]-1],0,1e9,k));
}
return 0;
}标签：inpu name std markdown follow query nod 就是 sdi
原文地址：http://www.cnblogs.com/DexterYsw/p/7954997.html
