【CF893F】Subtree Minimum Query-主席树

测试地址：Subtree Minimum Query
题目大意：给定一棵有根树，每个节点有一个权值，现在有若干个询问，每个询问给出x,k$x,k$，表示询问以x$x$为根的子树中，与x$x$的距离不超过k$k$的点中最小的权值。强制在线。
做法：本题需要用到主席树。
首先询问子树我们肯定想到将点按DFS序排序，然后每个询问就是要求点在DFS序中编号在[l,r]$[l,r]$，深度在[dep(x),dep(x)+k]$[dep(x),dep(x)+k]$内的所有点的最小权值。注意到DFS序中的[l,r]$[l,r]$内肯定不存在深度小于dep(x)$dep(x)$的点，所以问题转化为求点在DFS序中编号在[l,r]$[l,r]$，深度在[0,dep(x)+k]$[0,dep(x)+k]$内所有点的最小权值。那么我们按深度顺序建主席树，每次询问某一个版本内的区间最小值即可。
以下是本人代码：

#include <bits/stdc++.h>
#define inf 1000000001
using namespace std;
int n,r,m,a[100010],first[100010],tot=0,dep[100010],top[100010],q[100010],h,t,maxdep=0;
int rt[1000010],mn[20000010],ch[20000010][2],lft[100010],rht[100010],pos[100010],tim=0;
struct edge {int v,next;} e[200010];
bool vis[100010]={0};

void insert(int a,int b)
{
    e[++tot].v=b,e[tot].next=first[a],first[a]=tot;
}

void bfs()
{
    vis[r]=1;dep[r]=0;
    q[1]=r;h=t=1;
    while(h<=t)
    {
        int v=q[h++];
        top[dep[v]]=h-1;
        maxdep=max(maxdep,dep[v]);
        for(int i=first[v];i;i=e[i].next)
            if (!vis[e[i].v])
            {
                q[++t]=e[i].v;
                dep[e[i].v]=dep[v]+1;
                vis[e[i].v]=1;
            }
    }
}

void dfs(int v,int f)
{
    lft[v]=++tim;
    pos[tim]=v;
    for(int i=first[v];i;i=e[i].next)
        if (e[i].v!=f) dfs(e[i].v,v);
    rht[v]=tim;
}

void pushup(int x)
{
    mn[x]=min(mn[ch[x][0]],mn[ch[x][1]]);
}

void build(int &v,int l,int r)
{
    v=++tot;
    if (l==r) {mn[v]=inf;return;}
    int mid=(l+r)>>1;
    build(ch[v][0],l,mid);
    build(ch[v][1],mid+1,r);
    pushup(v);
}

void treeinsert(int &v,int last,int l,int r,int x)
{
    v=++tot;
    mn[v]=mn[last];
    ch[v][0]=ch[last][0],ch[v][1]=ch[last][1];
    if (l==r) {mn[v]=a[pos[x]];return;}
    int mid=(l+r)>>1;
    if (x<=mid) treeinsert(ch[v][0],ch[last][0],l,mid,x);
    else treeinsert(ch[v][1],ch[last][1],mid+1,r,x);
    pushup(v);
}

int treequery(int v,int l,int r,int s,int t)
{
    if (l>=s&&r<=t) return mn[v];
    int mid=(l+r)>>1,ans=inf;
    if (s<=mid) ans=min(ans,treequery(ch[v][0],l,mid,s,t));
    if (t>mid) ans=min(ans,treequery(ch[v][1],mid+1,r,s,t));
    return ans;
}

int main()
{
    scanf("%d%d",&n,&r);
    for(int i=1;i<=n;i++)
        scanf("%d",&a[i]);
    for(int i=1;i<n;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        insert(x,y),insert(y,x);
    }

    bfs();
    dfs(r,0);
    tot=0;
    build(rt[0],1,n);
    for(int i=1;i<=n;i++)
        treeinsert(rt[i],rt[i-1],1,n,lft[q[i]]);

    scanf("%d",&m);
    int last=0;
    while(m--)
    {
        int x,k;
        scanf("%d%d",&x,&k);
        x=(x+last)%n+1,k=(k+last)%n;
        last=treequery(rt[top[min(dep[x]+k,maxdep)]],1,n,lft[x],rht[x]);
        printf("%d
",last);
    }

    return 0;
}