[BZOJ1803][Spoj1487]Query on a tree III（dfs序+主席树）

最新推荐文章于 2018-08-09 16:28:00 发布

wwyx2001

最新推荐文章于 2018-08-09 16:28:00 发布

阅读量284

点赞数 1

CC 4.0 BY-SA版权

分类专栏：搜索（dfs序）主席树可持久化

本文链接：https://blog.youkuaiyun.com/Blue_CuSO4/article/details/78827125

搜索（dfs序）同时被 3 个专栏收录

32 篇文章

订阅专栏

主席树

17 篇文章

订阅专栏

可持久化

12 篇文章

订阅专栏

本文介绍了一种使用主席树和DFS序解决子树问题的方法。通过构建主席树并利用DFS序，可以有效地进行子树节点查询。文章提供了完整的代码实现，并详细解释了构建过程及查询操作。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目：

我是超链接

题解：

记得以前链剖求子树的时候都是dfs序，感觉这里也可以随便搞搞？
按照dfs序建立主席树然后跑就好啦
dfs序：

in[x]=++nn;
…遍历
out[x]=nn;

代码：

#include <map>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int N=1e5+5;
struct hh{int l,r,w;}t[N*20];
int tot,nxt[N*2],point[N],v[N*2],in[N],nn,out[N],sz,s,who[N],root[N],b[N],a[N],p[N];
map<int,int>mp;
void addline(int x,int y)
{
    ++tot; nxt[tot]=point[x]; point[x]=tot; v[tot]=y;
    ++tot; nxt[tot]=point[y]; point[y]=tot; v[tot]=x;
}
void dfs(int x,int fa)
{
    in[x]=++nn;who[nn]=x; 
    for (int i=point[x];i;i=nxt[i])
      if (v[i]!=fa) dfs(v[i],x);
    out[x]=nn;
}
void insert(int &now,int l,int r,int x)
{
    t[++sz]=t[now]; now=sz;
    t[now].w++;
    if (l==r) return;
    int mid=(l+r)>>1;
    if (x<=mid) insert(t[now].l,l,mid,x);
    else insert(t[now].r,mid+1,r,x);
}
int qurry(int i,int j,int l,int r,int k)
{
    if (l==r) return l;
    int mid=(l+r)>>1,sum=t[t[j].l].w-t[t[i].l].w;
    if (sum>=k) return qurry(t[i].l,t[j].l,l,mid,k);
    else return qurry(t[i].r,t[j].r,mid+1,r,k-sum);
}
int main()
{
    int n;
    scanf("%d",&n);
    for (int i=1;i<=n;i++) scanf("%d",&a[i]),b[++s]=a[i];
    sort(b+1,b+s+1);
    s=unique(b+1,b+s+1)-b-1;
    for (int i=1;i<=n;i++) p[i]=lower_bound(b+1,b+s+1,a[i])-b,mp[p[i]]=i;
    for (int i=1;i<n;i++)
    {
        int x,y; scanf("%d%d",&x,&y);
        addline(x,y);
    }
    dfs(1,0);
    for (int i=1;i<=n;i++)
    {
        root[i]=root[i-1];
        insert(root[i],1,s,p[who[i]]);
    }
    int q,x,k;
    scanf("%d",&q);
    while (q--)
    {
        scanf("%d%d",&x,&k);
        int l=in[x],r=out[x];
        printf("%d\n",mp[qurry(root[l-1],root[r],1,s,k)]);
    }
}