hdu 4718: The LCIS on the Tree

题目链接如下所示：

Problem - 4718l

理解一下树链剖分：

【洛谷日报#17】树链剖分详解 - 知乎

这里树链剖分的作用在于将这棵树切分成了离散的枝条，用这些离散的枝条构成了一颗线段树，对每一个离散区间进行区间的维护。

这里维护的方法可以参考hdu3308的做法。

不同之处在于，这里需要考虑方向，因为从树的一端到另外一端是有方向的，而像hdu3308这道就是没有方向，因为就是区间而已，而树链剖分从上到下的id是递增的，这说明我们需要记录一些方向信息并进行翻转才能得到正确方向。

代码如下所示：

#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
typedef long long ll;
const int MAXN=1e5+1;
int w[MAXN];// 各个顶点权值
int n,m;
vector<int> edge[MAXN];

int p[MAXN], dep[MAXN], sz[MAXN], son[MAXN];
void dfs1(int rt=1, int fa=0){
    p[rt]=fa, dep[rt]=dep[fa]+1, sz[rt]=1, son[rt]=0;
    for (auto &to : edge[rt]) {
        if (to==fa) continue;
        dfs1(to, rt);
        sz[rt]+=sz[to];
        if (sz[to]>sz[son[rt]]){
            son[rt]=to;
        }
    }
}

int id[MAXN], nw[MAXN], top[MAXN], ind;
void dfs2(int rt=1, int tp=1){
    id[rt]=++ind, nw[ind]=w[rt], top[rt]=tp;
    if (!son[rt]) return;
    dfs2(son[rt], tp);
    for (auto &to : edge[rt]) {
        if (to==son[rt] || to==p[rt]){
            continue;
        }
        dfs2(to, to);
    }
}


struct node{
    int l, r;
    int lmax,rmax,fmax;
    int LMAX,RMAX,FMAX;
    int vl, vr;

    void init(int L=0, int R=0){
        lmax=rmax=fmax=LMAX=RMAX=FMAX=vl=vr=0;
        l=L,r=R;
    }

    void reverse(){
        swap(lmax,RMAX), swap(rmax,LMAX),swap(fmax, FMAX);
        swap(vl,vr);
    }

}tree[MAXN<<2];

void push_up(node &rt, const node& l_rt, const node& r_rt){
    const int llen=l_rt.r-l_rt.l+1, rlen=r_rt.r-r_rt.l+1;
    rt.fmax = max({l_rt.fmax, r_rt.fmax, l_rt.vr<r_rt.vl ? l_rt.rmax+r_rt.lmax : 0});
    rt.lmax = l_rt.lmax + (l_rt.lmax==llen && l_rt.vr<r_rt.vl ? r_rt.lmax : 0);
    rt.rmax = r_rt.rmax + (r_rt.rmax==rlen && l_rt.vr<r_rt.vl ? l_rt.rmax : 0);

    rt.FMAX = max({l_rt.FMAX, r_rt.FMAX, l_rt.vr>r_rt.vl ? l_rt.RMAX+r_rt.LMAX : 0});
    rt.LMAX = l_rt.LMAX + (l_rt.LMAX==llen && l_rt.vr>r_rt.vl ? r_rt.LMAX : 0);
    rt.RMAX = r_rt.RMAX + (r_rt.RMAX==rlen && l_rt.vr>r_rt.vl ? l_rt.RMAX : 0);
    rt.vl=l_rt.vl, rt.vr=r_rt.vr;
}

void push_up(int rt){
    push_up(tree[rt], tree[rt<<1], tree[rt<<1|1]);
}

void build(int l,int r,int rt){
    tree[rt] = {l,r,1,1,1,1,1,1,nw[l],nw[l]};
    if (l>=r) return;
    int mid=(l+r)>>1;
    build(lson);
    build(rson);
    push_up(rt);
}

node query(int ul,int ur,int l,int r,int rt){
//    cout<<"interval [ul,ur]"<<"["<<ul<<","<<ur<<"], "<<""<<"[l,r]:["<<l<<","<<r<<"]"<<endl;
    if (ul <= l && r <= ur) return tree[rt];
    int mid = (l+r)>>1;
    if (ur<=mid) return query(ul, ur, lson);
    if (ul>mid) return query(ul,ur,rson);

    node left = query(ul,ur,lson);
    node right = query(ul,ur, rson);
    node res;
    res.init(left.l,right.r);
    push_up(res, left, right);
    return res;
}

node merge(const node& l, const node& r){
    if (!l.l) return r;
    if (!r.l) return l;
    node res;
    res.init(l.l,r.r);
    push_up(res,l,r);
    return res;
}

int query_route(int a,int b){
    node la,lb;
    la.init(),lb.init();
    int flag=0;
//    cout<<"route 1"<<endl;
    while (top[a]!=top[b]){
//        cout<<"dep:"<<dep[top[a]]<<", "<<dep[top[b]]<<endl;
        if (dep[top[a]] < dep[top[b]]) swap(a,b), swap(la,lb),flag^=1;
//        cout<<"id:"<<id[top[a]]<<", "<<id[a]<<endl;
        la = merge(query(id[top[a]], id[a],1,n,1), la);
        a=p[top[a]];
    }
//    cout<<"route 2"<<endl;
    if (id[a]>id[b]){
        swap(a,b), swap(la,lb),flag^=1;
    }
//    cout<<"route 3"<<endl;
    lb= merge(query(id[a],id[b],1,n,1),lb);
    if (flag) swap(la,lb);
    la.reverse();
    return merge(la,lb).fmax;
}

int main(){
    int kase;
    cin>>kase;
    for (int i = 1; i <= kase; ++i) {
        printf("Case #%d:\n", i); ind = 0;
        scanf("%d", &n);
        for (int i=1;i<=n;++i) {
            scanf("%d",&w[i]);
            edge[i].clear();
        }
        int father;
        for (int i = 2; i <= n; ++i) {
            scanf("%d",&father);
            edge[father].push_back(i);
        }

        dfs1();
        dfs2();
        build(1,n,1);
        scanf("%d", &m);
        int a,b;
        for (int i = 0; i < m; ++i) {
            scanf("%d %d",&a,&b);
//            cout<<"a:"<<a<<",b:"<<b<<endl;
            printf("%d\n", query_route(a,b));
        }
        if (i!=kase) puts("");
    }
    return 0;
}