BZOJ 1036 树链剖分

题意:之前写过lct版本的这个 就是点权 然后修改路径上的一个点的值 然后查询路径上的最大点权和路径权值之和

解法:这个是第一道树链剖分 和我所看的概念有点出入的 但是还是很容易可以得解的 

其核心思想就是利用dfs 然后把所有的路径进行划分 然后将所谓的重链划分在一起 然后采用线段树进行查询 和 修改


其实树链剖分还可以搞子树的 这一点和线段树转化序列相类似

感觉真的很不好 以前很早是时候在子树查询的时候就已经接触到过了 现在才学习这个 其实也真的很简单 就是自己被吓到了而已  如果偏安一隅的话 就没有必要来搞竞赛了

#include<cstdio>
#include<cstring>
#include<limits.h>
#include<string.h>
#include<algorithm>
#define inf INT_MAX
#define maxn 33333
#define ls (rt<<1)
#define rs (rt<<1|1)
#define mid ((l+r)>>1)
using namespace std;
int fa[maxn][15],v[maxn],deep[maxn],son[maxn],pl[maxn],belong[maxn];
int x,y,cnt,sz,q,n;
bool vis[maxn];
struct data{int to,next;}e[maxn<<1];
int head[maxn];
struct seg{int l,r,mx,sum;}t[100001];
inline void add_e(int u,int v){
    e[++cnt].to=v;e[cnt].next=head[u];head[u]=cnt;
    e[++cnt].to=u;e[cnt].next=head[v];head[v]=cnt;
}
inline void dfs1(int u){son[u]=vis[u]=1;
    for(int i=1;i<=14&&deep[u]>=(1<<i);++i)fa[u][i]=fa[fa[u][i-1]][i-1];
    for(int i=head[u];i;i=e[i].next){
        int v=e[i].to;if(vis[v])continue;
        deep[v]=deep[u]+1;fa[v][0]=u;
        dfs1(v); son[u]+=son[v];
    }
}
inline void dfs2(int u,int chain){
    int k=0;sz++;pl[u]=sz,belong[u]=chain;
    for(int i=head[u];i;i=e[i].next){
        int v=e[i].to;
        if(deep[v]>deep[u]&&son[v]>son[k])k=e[i].to;
    }
    if(!k)return ;
    dfs2(k,chain);
    for(int i=head[u];i;i=e[i].next){
        int v=e[i].to;
        if(deep[v]>deep[u]&&k!=v)dfs2(v,v);
    }
}

inline int lca(int x,int y){
    if(deep[x]<deep[y])swap(x,y);
    int t=deep[x]-deep[y];
    for(int i=0;i<=14;++i)if(t&(1<<i))x=fa[x][i];
    for(int i=14;i>=0;i--)if(fa[x][i]!=fa[y][i])x=fa[x][i],y=fa[y][i];
    return x==y?x:fa[x][0];
}
inline void build(int rt,int l,int r){
    t[rt].l=l,t[rt].r=r;
    if(l==r)return ;
    build(ls,l,mid);build(rs,mid+1,r);
}
inline void change(int rt,int pos,int w){
    int l=t[rt].l,r=t[rt].r;
    if(l==r){t[rt].sum=t[rt].mx=w;return;}
    if(pos<=mid)change(ls,pos,w);
    if(mid<pos)change(rs,pos,w);
    t[rt].sum=t[ls].sum+t[rs].sum;
    t[rt].mx=max(t[ls].mx,t[rs].mx);
}
inline int _sum(int rt,int L,int R){
    int l=t[rt].l,r=t[rt].r;
    if(L==l&&r==R)return t[rt].sum;
    if(R<=mid)return _sum(ls,L,R);
    if(mid<L)return _sum(rs,L,R);
    else return _sum(ls,L,mid)+_sum(rs,mid+1,R);
}
int _mx(int rt,int L,int R){
    int l=t[rt].l,r=t[rt].r;
    if(L==l&&r==R)return t[rt].mx;
    if(R<=mid)return _mx(ls,L,R);
    else if(L>mid)return _mx(rs,L,R);
    else return max(_mx(ls,L,mid),_mx(rs,mid+1,R));
}
inline int querys(int rt,int f){
    int sum=0;
    while(belong[rt]!=belong[f]){
        sum+=_sum(1,pl[belong[rt]],pl[rt]);
        rt=fa[belong[rt]][0];
    }sum+=_sum(1,pl[f],pl[rt]);
    return sum;
}
inline int querym(int rt,int f){
    int mx=-inf;
    while(belong[rt]!=belong[f]){
        mx=max(mx,_mx(1,pl[belong[rt]],pl[rt]));
        rt=fa[belong[rt]][0];
    }mx=max(mx,_mx(1,pl[f],pl[rt]));
    return mx;
}
char op[111];
int main(){
    scanf("%d",&n);
    for(int i=1;i<n;++i){
        scanf("%d%d",&x,&y);
        add_e(x,y);
    }
    for(int i=1;i<=n;++i)scanf("%d",&v[i]);
    
    dfs1(1);
    dfs2(1,1);
    
    build(1,1,n);
    for(int i=1;i<=n;++i)change(1,pl[i],v[i]);
    scanf("%d",&q);
    
    for(int i=1;i<=q;i++){
        int x,y;scanf("%s%d%d",op,&x,&y);
        if(*op=='C'){v[x]=y;change(1,pl[x],y);}
        else{
            int t=lca(x,y);
            if(op[1]=='M')printf("%d\n",max(querym(x,t),querym(y,t)));
            else printf("%d\n",querys(x,t)+querys(y,t)-v[t]);
        }
    }
    return 0;
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值