COJ 1011 WZJ的数据结构（十一）树上k大

题解：主席树&DFS序。

PS：为什么我一开始Wa了N发 是因为有一个左区间我写成[L,M+1]了。。。。。。。。。。。。。。。。。。。。。。。。。。

 #include<iostream>
 #include<cstdio>
 #include<cmath>
 #include<algorithm>
 #include<queue>
 #include<cstring>
 #define PAU putchar(' ')
 #define ENT putchar('\n')
 using namespace std;
 const int maxn=+,maxnode=+,maxn2=+;
 int ls[maxnode],rs[maxnode],s[maxnode],A[maxn],root[maxn],p[maxn],si[maxn],so[maxn],siz[maxn],cz=,tot=;
 struct tedge{int x,y,next;}adj[maxn2];int ms=,fch[maxn];
 void addedge(int u,int v){
     adj[++ms]=(tedge){u,v,fch[u]};fch[u]=ms;
     adj[++ms]=(tedge){v,u,fch[v]};fch[v]=ms;
     return;
 }
 void build(int x,int&y,int L,int R,int pos){
     s[y=++tot]=s[x]+;if(L==R) return;
     int M=L+R>>;     if(pos<=M) rs[y]=rs[x],build(ls[x],ls[y],L,M,pos);
     else ls[y]=ls[x],build(rs[x],rs[y],M+,R,pos);
 }
 int query(int x,int y,int L,int R,int k){
     if(L==R) return L;int M=L+R>>,kth=s[ls[y]]-s[ls[x]];
     if(k<=kth) return query(ls[x],ls[y],L,M,k);
     else return query(rs[x],rs[y],M+,R,k-kth);
 }
 void dfs(int u,int fa){
     si[u]=++cz;p[cz]=u;siz[u]=;
     for(int i=fch[u];i;i=adj[i].next){
         int v=adj[i].y;
         if(v!=fa) dfs(v,u),siz[u]+=siz[v];
     } so[u]=cz;return;
 }
 inline int read(){
     int x=,sig=;char ch=getchar();
     while(!isdigit(ch)){if(ch=='-')sig=-;ch=getchar();}
     while(isdigit(ch))x=*x+ch-'',ch=getchar();
     return x*=sig;
 }
 inline void write(int x){
     if(x==){putchar('');return;}if(x<)putchar('-'),x=-x;
     int len=,buf[];while(x)buf[len++]=x%,x/=;
     for(int i=len-;i>=;i--)putchar(buf[i]+'');return;
 }
 int n,Q;
 void init(){
     n=read();Q=read();
     for(int i=;i<n;i++) addedge(read(),read());
     for(int i=;i<=n;i++) A[i]=read();
     dfs(,);
     for(int i=;i<=n;i++) build(root[i-],root[i],,n,A[p[i]]);
     return;
 }
 void work(){
     int x,k;
     while(Q--){
         x=read();k=read();
         if(siz[x]<k) puts("-1");
         else write(query(root[si[x]-],root[so[x]],,n,k)),ENT;
     }
     return;
 }
 void print(){
     return;
 }
 int main(){init();work();print();return ;}

搜索

复制