1803: Spoj1487 Query on a tree III

题目链接

题目大意：子树k大

题解：dfs序主席树

我的收获：主席树吼啊

#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;

#define M 100005
#define MX M*20
#define lson tl[x],tl[y],l,mid
#define rson tr[x],tr[y],mid+1,r

int n,m,t,tot,cnt,tim;
int root[M],a[M],num[M],head[M];
int dfn[M],out[M],pre[M],mp[M];
int tl[MX],tr[MX],sum[MX];

struct edge{int to,nex;}e[M*2];

inline void add(int u,int v){e[t].to=v,e[t].nex=head[u],head[u]=t++;}

inline void node(int &x,int w){x=++cnt,sum[x]=sum[w]+1,tl[x]=tl[w],tr[x]=tr[w];}

void insert(int x,int &y,int l,int r,int k)
{
    node(y,x);
    if(l==r) return ;
    int mid=(l+r)>>1;
    if(k<=mid) insert(lson,k);
    else insert(rson,k);
}

int query(int x,int y,int l,int r,int k)
{
    if(l==r) return l;
    int mid=(l+r)>>1,t=sum[tl[y]]-sum[tl[x]];
    if(t>=k) return query(lson,k);
    return query(rson,k-t);
}

void dfs(int x,int fa){
    dfn[x]=++tim;pre[dfn[x]]=x;
    for(int i=head[x];i!=-1;i=e[i].nex)
    if(e[i].to!=fa) dfs(e[i].to,x);
    out[x]=tim;
}

void work()
{
    for(int x,z,i=1;i<=m;i++){
        scanf("%d%d",&x,&z);
        printf("%d\n",mp[query(root[dfn[x]-1],root[out[x]],1,tot,z)]);
    }
}

void lisan()
{
    sort(num+1,num+n+1);
    tot=unique(num+1,num+n+1)-(num+1);
    for(int i=1;i<=n;i++) a[i]=lower_bound(num+1,num+tot+1,a[i])-num,mp[a[i]]=i;
}

void init()
{
    scanf("%d",&n);
    memset(head,-1,sizeof(head));
    for(int i=1;i<=n;i++) scanf("%d",&a[i]),num[i]=a[i];
    lisan();
    for(int x,y,i=1;i<n;i++) scanf("%d%d",&x,&y),add(x,y),add(y,x);
    dfs(1,0);
    scanf("%d",&m);
    for(int i=1;i<=n;i++) insert(root[i-1],root[i],1,tot,a[pre[i]]);
}

int main()
{
    init();
    work();
}