[LCT]SCOI2018：tree

最新推荐文章于 2019-10-27 22:20:24 发布

romiqi_new

最新推荐文章于 2019-10-27 22:20:24 发布

阅读量301

点赞数

分类专栏： LCT

本文链接：https://blog.csdn.net/qq_43346903/article/details/97523014

版权

LCT 专栏收录该内容

18 篇文章 0 订阅

订阅专栏

Description：

在大小为 N 的树上，点从 1 到 N 标号，第 i 个点有权值 Ai，现在需要支持两种操作：
第一种操作格式为“1 U” ，表示询问从 U 出发的简单路径，经过的点权值之和的最大值；
第二种操作格式为“2 U V” ，表示将 U 的权值修改为 V。

Data Range：

对于 10%的数据，满足：1<=N<=1000,1<=M<=1000
对于另外 20%的数据，不存在修改操作
对于另外 20%的数据，满足 Pi=i
对于 100%的数据，满足：1<=N<=100000， 1<=M<=100000，1<=Pi<=i，
-10000<=Ai<=10000，1<=U<=n，-10000<=V<=10000

Solution：

类似Qtree，每个点维护一个multiset表示其虚子树的路径权值和最大值
然后实链维护上下在当前链上的路径权值和以及链的所有虚子树权值和，就好修改了
询问就把一个点变成链底，直接查询就完了

Code：

#include<bits/stdc++.h>
using namespace std;
inline int read(){
    int res=0,f=1;char ch=getchar();
    while(!isdigit(ch)) {if(ch=='-') f=-f;ch=getchar();}
    while(isdigit(ch)) {res=(res<<1)+(res<<3)+(ch^48);ch=getchar();}
    return res*f;
}
const int INF=1e9,N=1e5+5;
int vis[N<<1],head[N<<1],nxt[N<<1],fr[N<<1],tot=0;
inline void add(int x,int y){vis[++tot]=y;nxt[tot]=head[x];head[x]=tot;fr[tot]=x;}
multiset<int,greater<int> >s[N];
multiset<int,greater<int> >::iterator it;
int get(int x){
    if(!s[x].size()) return -INF;
    else return (*s[x].begin());
}
namespace LCT{ 
    int ls[N],rs[N],fa[N],siz[N],val[N],lmx[N],rmx[N];
    inline void pushup(int x){
        siz[x]=val[x];
        if(ls[x]) siz[x]+=siz[ls[x]];
        if(rs[x]) siz[x]+=siz[rs[x]];
        lmx[x]=max(lmx[ls[x]],siz[ls[x]]+max(val[x],max(get(x),lmx[rs[x]])+val[x]));
        rmx[x]=max(rmx[rs[x]],siz[rs[x]]+max(val[x],max(get(x),rmx[ls[x]])+val[x]));
    }
    inline int isrs(int x){return rs[fa[x]]==x;}
    inline bool isroot(int x){
        if(!fa[x]) return 1;
        return ls[fa[x]]!=x && rs[fa[x]]!=x;
    }
    inline void rotate(int x){
        int y=fa[x],z=fa[y],b=ls[y]==x?rs[x]:ls[x];
        if(z && !isroot(y)) (ls[z]==y?ls[z]:rs[z])=x;
        fa[x]=z,fa[y]=x;b?fa[b]=y:0;
        if(ls[y]==x) rs[x]=y,ls[y]=b;
        else ls[x]=y,rs[y]=b;
        pushup(y);pushup(x);
    }
    inline void splay(int x){
        while(!isroot(x)){
            if(!isroot(fa[x])){
                if(isrs(x)==isrs(fa[x])) rotate(fa[x]);
                else rotate(x);
            }
            rotate(x);
        }
        pushup(x);
    }
    inline void access(int x){
        for(int y=0;x;y=x,x=fa[x]){
            splay(x);
            if(rs[x]) s[x].insert(lmx[rs[x]]);
            if(y){
                it=s[x].find(lmx[y]);
                s[x].erase(it);
            }
         rs[x]=y;pushup(x);
        }
    }
}
using namespace LCT;
void dfs(int v,int f){
    siz[v]=ls[v]=rs[v]=val[v]=lmx[v]=rmx[v]=0;
    fa[v]=f;s[f].insert(0);
    for(int i=head[v];i;i=nxt[i]) dfs(vis[i],v);
}
int main(){
    lmx[0]=rmx[0]=-INF;
    int n=read(),q=read();
    for(int x,i=2;i<=n;i++){x=read();add(x,i);}
    dfs(1,0);
    for(int i=1;i<=n;i++){access(i);splay(i);val[i]=read();pushup(i);}
    while(q--){
        int op=read();
        if(op==1){
            int x=read();
            access(x);splay(x);cout<<rmx[x]<<"\n";
        }
        else{
            int x=read(),y=read();
            access(x);splay(x);val[x]=y;pushup(x);
        }
    }
    return 0;
}