[vijos]lxhgww的奇思妙想——长链剖分模板

题目大意：

给定一棵树，询问x的k次祖先。

思路：

这是一道长链剖分的模板。
长链剖分类似于平常树链剖分的模式，但是划分轻重链并不是以size为依据而是以子树内最深度最大的节点为依据，这使得它具有一些重链剖分不具有的性质。
性质1：所有的链长=n
性质2：任意一个点的k次祖先所在的链链长\(\geq\)k。
上面两条性质都很好证，这里不再作多余说明。
于是考虑如何高效地求解k次祖先，我们在每条链的端点记录两个表，一个表示向下的点有哪些，一个表示向上的链长个点有哪些。
询问时我们可以将k折半，x首先跳到x的\(\frac{k}{2}\)次祖先y，不难发现这时y所在的链长\(\geq \frac{k}{2}\)，同时它距离x的k次祖先的距离也\(\leq \frac{k}{2}\)，不难发现这个时候无论y和x的k次祖先在不在一条链上，都可以通过向上或者向下的数组跳到。
这里的折半操作在实际操作中是利用预处理的倍增数组跳到x的\(2^{\lfloor{log_2k\rfloor}}\)次祖先处，不难发现此时单次询问复杂度O(1)。

#include<bits/stdc++.h>

#define REP(i,a,b) for(int i=a,i##_end_=b;i<=i##_end_;++i)
#define DREP(i,a,b) for(int i=a,i##_end_=b;i>=i##_end_;--i)
#define debug(x) cout<<#x<<"="<<x<<" "
#define siz(x) ((int)x.size())
#define fi first
#define se second
#define mk make_pair
#define pb push_back
typedef long long ll;

using namespace std;

void File(){
    freopen("lxhgww.in","r",stdin);
    freopen("lxhgww.out","w",stdout);
}

template<typename T>void read(T &_){
    _=0; T f=1; char c=getchar();
    for(;!isdigit(c);c=getchar())if(c=='-')f=-1;
    for(;isdigit(c);c=getchar())_=(_<<1)+(_<<3)+(c^'0');
    _*=f;
}

const int maxn=3e5+10;
int n,m,Log[maxn],st[maxn][21];
int beg[maxn],to[maxn<<1],las[maxn<<1],cnte=1;
int fa[maxn],son[maxn],len[maxn],top[maxn],dep[maxn];
vector<int>lis[maxn],up[maxn];

void add(int u,int v){
    las[++cnte]=beg[u]; beg[u]=cnte; to[cnte]=v;
    las[++cnte]=beg[v]; beg[v]=cnte; to[cnte]=u;
}

void dfs1(int u,int f){
    fa[u]=f;
    st[u][0]=f;
    dep[u]=dep[f]+1;
    for(int i=beg[u];i;i=las[i]){
        int v=to[i];
        if(v==f)continue;
        dfs1(v,u);
        if(len[v]+1>len[u]){
            len[u]=len[v]+1;
            son[u]=v;
        }
    }
}

void dfs2(int u,int t){
    top[u]=t;
    lis[t].pb(u);
    if(son[u])dfs2(son[u],t);
    for(int i=beg[u];i;i=las[i]){
        int v=to[i];
        if(v==fa[u] || v==son[u])continue;
        dfs2(v,v);
    }
    /*if(u==t){
        int p=u;
        REP(i,1,len[u]){
            if(!p)break;
            up[u].pb(p);
            p=fa[p];
        }
    }*/
}

void init(){
    read(n);
    REP(i,2,n)Log[i]=Log[i/2]+1;
    int u,v;
    REP(i,1,n-1)read(u),read(v),add(u,v);
    dfs1(1,0);
    dfs2(1,1);
    for(u=1;u<=n;++u)
        if(u==top[u])for(int i=1,p=u;i<=len[u] && p;++i,p=fa[p])
            up[u].pb(p);
    REP(j,1,20)REP(i,1,n)
        if((1<<j)<=dep[i]-1)
            st[i][j]=st[st[i][j-1]][j-1];
}

int query(int u,int k){
    if(k>dep[u]-1)return 0;
    if(!k)return u;
    u=st[u][Log[k]];
    k-=1<<Log[k];
    int d=dep[u]-dep[top[u]];
    if(k<=d)return lis[top[u]][d-k];
    else return up[top[u]][k-d];
}

void work(){
    read(m);
    int x,y,ans=0;
    REP(i,1,m){
        read(x),read(y);
        x^=ans; y^=ans;
        ans=query(x,y);
        printf("%d\n",ans);
    }
}

int main(){
    //File();
    init();
    work();
    return 0;
}

转载于:https://www.cnblogs.com/ylsoi/p/10234313.html