bzoj4551: [Tjoi2016&Heoi2016]樹（樹鏈剖分）

題目傳送門
。

解法：
樹剖套線段樹。
維護區間最右標記。

程式碼實現：

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
using namespace std;
struct node {int x,y,next;}a[210000];int len,last[110000];
void ins(int x,int y) {len++;a[len].x=x;a[len].y=y;a[len].next=last[x];last[x]=len;}
int fa[110000],tot[110000],son[110000],dep[110000];
void pre_tree_node(int x) {
    tot[x]=1;son[x]=0;
    for(int k=last[x];k;k=a[k].next) {
        int y=a[k].y;
        if(fa[x]!=y) {
            dep[y]=dep[x]+1;fa[y]=x;pre_tree_node(y);
            if(tot[son[x]]<tot[y])son[x]=y;tot[x]+=tot[y];
        }
    }
}
int z,ys[110000],top[110000],dian[110000];
void pre_tree_edge(int x,int tp) {
    top[x]=tp;ys[x]=++z;dian[z]=x;
    if(son[x]!=0)pre_tree_edge(son[x],tp);
    for(int k=last[x];k;k=a[k].next) {
        int y=a[k].y;
        if(y!=fa[x]&&y!=son[x])pre_tree_edge(y,y);
    }
}
struct trnode {int l,r,lc,rc,c;}tr[410000];int trlen;
void bt(int l,int r) {
    trlen++;int now=trlen;
    tr[now].l=l;tr[now].r=r;tr[now].c=0;tr[now].rc=tr[now].lc=-1;
    if(l<r) {
        int mid=(l+r)/2;tr[now].lc=trlen+1;bt(l,mid);
        tr[now].rc=trlen+1;bt(mid+1,r);
    }
}
void change(int now,int x) {
    if(tr[now].l==tr[now].r){tr[now].c=x;return ;}
    int lc=tr[now].lc,rc=tr[now].rc,mid=(tr[now].l+tr[now].r)/2;
    if(x<=mid)change(lc,x);else change(rc,x);
    tr[now].c=max(tr[lc].c,tr[rc].c);
}
int find(int now,int l,int r) {
    if(tr[now].l==l&&tr[now].r==r)return tr[now].c;
    int lc=tr[now].lc,rc=tr[now].rc,mid=(tr[now].l+tr[now].r)/2;
    if(r<=mid)return find(lc,l,r);else if(l>mid)return find(rc,l,r);
    else {
        int tt=find(rc,mid+1,r);if(tt!=0)return tt;
        return find(lc,l,mid);
    }
}
int solve(int x) {
    int tx=top[x];
    while(tx!=1) {
        int tt=find(1,ys[tx],ys[x]);if(tt!=0)return dian[tt];
        x=fa[tx];tx=top[x];
    }
    return dian[find(1,ys[1],ys[x])];
}
int main() {
    int n,m;scanf("%d%d",&n,&m);len=0;memset(last,0,sizeof(last));
    for(int i=1;i<n;i++) {int x,y;scanf("%d%d",&x,&y);ins(x,y);ins(y,x);}
    fa[1]=0;dep[1]=0;pre_tree_node(1);z=0;pre_tree_edge(1,1);trlen=0;bt(1,z);change(1,ys[1]);
    for(int i=1;i<=m;i++) {
        char s[5];int x;scanf("%s%d",s+1,&x);
        if(s[1]=='Q')printf("%d\n",solve(x));
        else change(1,ys[x]);
    }
    return 0;
}