CodeForces 1062E Company (主席树第k大+dfs序性质)

#include<bits/stdc++.h>
using namespace std;
#define debug puts("YES");
#define rep(x,y,z) for(int (x)=(y);(x)<(z);(x)++)
#define ll long long
#define lrt int l,int r,int rt
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define mst(a,b) memset((a),(b),sizeof(a))
#define pii pair<int,int>
#define fi first                                  
#define se second
#define mk(x,y) make_pair(x,y)
const int mod=1e9+7;
const int maxn=1e5+100;
const int maxm=5e5+10;
const ll INF=1e18;
ll powmod(ll x,ll y){ll t; for(t=1;y;y>>=1,x=x*x%mod) if(y&1) t=t*x%mod; return t;}
ll gcd(ll x,ll y){
    if(y==0) return x;
    return gcd(y,x%y);
}
int n,q;
vector<int> g[maxn];
int fa[maxn][20],dep[maxn],dfn[maxn],mp[maxn],tot=0;
void dfs(int u,int f){
    dep[u]=dep[f]+1,fa[u][0]=f,dfn[u]=++tot,mp[tot]=u;
    rep(i,1,20) fa[u][i]=fa[fa[u][i-1]][i-1];
    for(auto v:g[u]) if(v^f)
        dfs(v,u);
}
///主席树
int cnt=0;
int root[maxn],ch[maxn*20][2],val[maxn*20];
void update(int old,int &cur,int l,int r,int pos){
    cur=++cnt;
    ch[cur][0]=ch[old][0],ch[cur][1]=ch[old][1];
    val[cur]=val[old]+1;
    if(l==r) return ;
    int mid=l+r>>1;
    if(pos<=mid) update(ch[old][0],ch[cur][0],l,mid,pos);
    else update(ch[old][1],ch[cur][1],mid+1,r,pos);
}
int query(int old,int cur,int l,int r,int k){
    if(l==r) return mp[l];
    int tmp=val[ch[cur][0]]-val[ch[old][0]];
    int mid=l+r>>1;
    if(tmp>=k)
        return query(ch[old][0],ch[cur][0],l,mid,k);
    else 
        return query(ch[old][1],ch[cur][1],mid+1,r,k-tmp);
}
int LCA(int x,int y){
    if(dep[x]<dep[y]) swap(x,y);
    int tmp=dep[x]-dep[y];
    int i=0;
    while((1<<i)<tmp) i++;
    for(;i>=0;i--) if(tmp&(1<<i))
        x=fa[x][i];
    if(x==y) return x;
    for(i=19;i>=0;i--)  if(fa[x][i]^fa[y][i])
        x=fa[x][i],y=fa[y][i];
    return fa[x][0];
}
int main(){
    ios::sync_with_stdio(false);/*
    freopen("d://in.txt","r",stdin);
    freopen("d://out.txt","w",stdout);*/
    cin>>n>>q;
    rep(i,2,n+1){
        int x;cin>>x;
        g[x].push_back(i);
        g[i].push_back(x);
    }
    dfs(1,1);
    rep(i,1,n+1){
        update(root[i-1],root[i],1,n,dfn[i]);
    }
    rep(i,0,q){
        int x,y;
        cin>>x>>y;
        int tmp=y-x+1,tx,ty;
        int ans1,ans2,ans3,ans4;
        int tp1=query(root[x-1],root[y],1,n,1),tp2=query(root[x-1],root[y],1,n,tmp);
        ans1=LCA(tp1,query(root[x-1],root[y],1,n,tmp-1));
        ans2=dep[ans1]-1;
        ans3=LCA(query(root[x-1],root[y],1,n,2),tp2);
        ans4=dep[ans3]-1;
        if(ans2>ans4) cout<<tp2<<" "<<ans2<<"\n";
        else cout<<tp1<<" "<<ans4<<"\n";
    }
    return 0;
}

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值