Codeforces893 F. Subtree Minimum Query(线段树合并)

题意:

给定一棵n个点的树,每个点有点权a(i),每条边的边权为1
m次询问,每次询问给出(x,k),
表示查询以x为根的子树中,到达x的距离小于等于k的所有点中的最小点权是多少
询问强制在线

数据范围:n<=1e5

解法:

设第i个节点的深度是d(i)
对于x子树中的每个节点,记录深度=k时候点权的最小值min(k)
那么查询距离x不超过k的点的最小值,就是在min(d[x])…min(d[x+k])中查询最小值,可以用线段树来搞

每个节点都对应一棵线段树,且每个节点的线段树都可以由子节点的线段树合并得来,所以可以用线段树合并来做

自底向上建树,记录每个节点对应的线段树根节点,询问的时候直接查询即可。

code:
#include<bits/stdc++.h>
using namespace std;
const int maxm=2e5+5;
vector<int>g[maxm];
int a[maxm];
int n,rt;
int ans;
//
int d[maxm];
int lc[maxm*40],rc[maxm*40],mi[maxm*40];
int r[maxm],tot;
//
void pushup(int k){
    mi[k]=min(mi[lc[k]],mi[rc[k]]);//lc[k]和rc[k]可能为0,所以程序一开始需要初始化mi[0]=inf
}
void update(int x,int val,int l,int r,int &k){//tree[x]=val
    if(!k)k=++tot;
    if(l==r){
        mi[k]=val;return ;
    }
    int mid=(l+r)/2;
    if(x<=mid)update(x,val,l,mid,lc[k]);
    else update(x,val,mid+1,r,rc[k]);
    pushup(k);
}
int merged(int x,int y){
    if(!x||!y)return x+y;
    int k=++tot;
    lc[k]=merged(lc[x],lc[y]);
    rc[k]=merged(rc[x],rc[y]);
    mi[k]=min(mi[x],mi[y]);
    return k;
}
int ask(int st,int ed,int l,int r,int k){
    if(st<=l&&ed>=r)return mi[k];
    int mid=(l+r)/2;
    int ans=2e9;
    if(st<=mid)ans=min(ans,ask(st,ed,l,mid,lc[k]));
    if(ed>mid)ans=min(ans,ask(st,ed,mid+1,r,rc[k]));
    return ans;
}
void dfs(int x,int fa){
    update(d[x],a[x],1,n,r[x]);
    for(int v:g[x]){
        if(v==fa)continue;
        d[v]=d[x]+1;
        dfs(v,x);
        r[x]=merged(r[x],r[v]);
    }
}
signed main(){
    //init
    mi[0]=2e9;
    //
    scanf("%d%d",&n,&rt);
    for(int i=1;i<=n;i++)scanf("%d",&a[i]);
    for(int i=1;i<n;i++){
        int a,b;scanf("%d%d",&a,&b);
        g[a].push_back(b);
        g[b].push_back(a);
    }
    d[rt]=1;
    dfs(rt,rt);
    int m;
    scanf("%d",&m);
    while(m--){
        int p,q;scanf("%d%d",&p,&q);
        int x=(p+ans)%n+1,k=(q+ans)%n;
        ans=ask(d[x],min(n,d[x]+k),1,n,r[x]);//注意有边界要取min(n,d[x]+k)
        printf("%d\n",ans);
    }
    return 0;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值