CF893F:Subtree Minimum Query(线段树合并)

Description

给你一颗有根树，点有权值，m次询问，每次问你某个点的子树中距离其不超过k的点的权值的最小值。（边权均为1，点权有可能重复，k值每次询问有可能不同，强制在线）

Input

第一行两个数，为点数$n$和树根$r$。

第二行$n$个数，为每个点的权值。

后面$n-1$行给出树边

再一行$m$，后面给出$m$组询问。

Output

一行答案。

Sample Input

5 2

1 3 2 3 5

2 3

5 1

3 4

4 1

2

1 2

2 3

Sample Output

2

5

Solution

线段树合并板子题。注意数组大小QAQ我老是开小$RE$……

感觉$Slr$和$beretty$两个人数据结构学傻了用了另一个题非常麻烦的做法写的这个题……

Code

#include<iostream>
#include<cstdio>
#define N (100009)
using namespace std;

struct Sgt{int ls,rs,min;}Segt[N<<6];
struct Edge{int to,next;}edge[N<<1];
int n,m,sgt_num,ans,lastans,u,v,p,q,r;
int a[N],Root[N],Depth[N];
int head[N],num_edge;

void add(int u,int v)
{
    edge[++num_edge].to=v;
    edge[num_edge].next=head[u];
    head[u]=num_edge;
}

void Update(int &now,int l,int r,int x,int v)
{
    if (!now) now=++sgt_num;
    Segt[now].min=2e9;
    if (l==r) {Segt[now].min=v; return;}
    int mid=(l+r)>>1;
    if (x<=mid) Update(Segt[now].ls,l,mid,x,v);
    else Update(Segt[now].rs,mid+1,r,x,v);
    int ls=Segt[now].ls, rs=Segt[now].rs;
    Segt[now].min=min(Segt[ls].min,Segt[rs].min);
}

int Merge(int x,int y)
{
    if (!x || !y) return x|y;
    int tmp=++sgt_num;
    Segt[tmp].ls=Merge(Segt[x].ls,Segt[y].ls);
    Segt[tmp].rs=Merge(Segt[x].rs,Segt[y].rs);
    Segt[tmp].min=min(Segt[x].min,Segt[y].min);
    return tmp;
}

void DFS(int x,int fa)
{
    Depth[x]=Depth[fa]+1;
    Update(Root[x],1,n,Depth[x],a[x]);
    for (int i=head[x]; i; i=edge[i].next)
        if (edge[i].to!=fa)
        {
            DFS(edge[i].to,x);
            Root[x]=Merge(Root[x],Root[edge[i].to]);
        }
}

int Query(int now,int l,int r,int l1,int r1)
{
    if (l>r1 || r<l1) return 2e9;
    if (l1<=l && r<=r1) return Segt[now].min;
    int mid=(l+r)>>1, ls=Segt[now].ls, rs=Segt[now].rs;
    return min(Query(ls,l,mid,l1,r1),Query(rs,mid+1,r,l1,r1));
}

int main()
{
    Segt[0].min=2e9;
    scanf("%d%d",&n,&r);
    for (int i=1; i<=n; ++i)
        scanf("%d",&a[i]);
    for (int i=1; i<=n-1; ++i)
    {
        scanf("%d%d",&u,&v);
        add(u,v); add(v,u);
    }
    DFS(r,0);
    scanf("%d",&m);
    for (int i=1; i<=m; ++i)
    {
        scanf("%d%d",&p,&q);
        p=(p+lastans)%n+1; q=(q+lastans)%n;
        ans=Query(Root[p],1,n,Depth[p],Depth[p]+q);
        printf("%d
",ans); lastans=ans;
    }
}