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,rint 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();}