树链剖分——BZOJ1036/Luogu2590 [ZJOI2008]树的统计

题面:BZOJ1036 Luogu2590
树剖裸题,比luogu某模板题不知道到哪里去了。。。
今天刚入门,看到一篇很好的树剖入门教程,写得很不错:传送门
对于初学者来说讲得很详细
最后来说说这道题
单点修改直接暴力找进线段树修改
求链上最大以及求和同链上修改,只不过不必更新权值
打得谨慎一点就行了
献上我的丑陋程序+注释

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<iostream>
#include<cstdlib>
#include<string>
#include<ctime>
#include<queue>
#include<climits>
using namespace std;
char c[10];
int n,nedge=0,v[100001],p[200001],nex[200001],head[200001];
int fa[100001],deep[100001],s[100001],son[100001],top[100001];
int sx[100001],xs[100001],ne=0;
int lt[400001],rt[400001],t[400001],mt[400001];
inline void addedge(int a,int b){p[++nedge]=b;nex[nedge]=head[a];head[a]=nedge;}
inline void dfs(int x,int Fa,int dep){//第一次dfs求父亲,子树数目,重儿子,深度
    fa[x]=Fa;deep[x]=dep;s[x]=1;
    for(int k=head[x];k;k=nex[k]){
        if(p[k]==Fa)continue;
        dfs(p[k],x,dep+1);s[x]+=s[p[k]];
        if(!son[x]||s[p[k]]>s[son[x]])son[x]=p[k];
    }
}
inline void dfss(int x,int bh){//第二次dfs求在线段树内位置,重链最顶端位置
    top[x]=bh;sx[x]=++ne;xs[sx[x]]=x;
    if(!son[x])return;
    dfss(son[x],bh);
    for(int k=head[x];k;k=nex[k])if(p[k]!=son[x]&&p[k]!=fa[x])dfss(p[k],p[k]);
}
inline void build(int l,int r,int nod){//建线段树
    lt[nod]=l;rt[nod]=r;
    if(l==r){t[nod]=mt[nod]=v[xs[l]];return;}
    int mid=l+r>>1;
    build(l,mid,nod*2);build(mid+1,r,nod*2+1);
    t[nod]=t[nod*2]+t[nod*2+1];mt[nod]=max(mt[nod*2],mt[nod*2+1]);
}
inline void xg(int x,int nod,int w){//线段树修改
    if(lt[nod]==rt[nod]){t[nod]=mt[nod]=w;return;}
    int mid=lt[nod]+rt[nod]>>1;
    if(x<=mid)xg(x,nod*2,w);
    else xg(x,nod*2+1,w);
    t[nod]=t[nod*2]+t[nod*2+1];mt[nod]=max(mt[nod*2],mt[nod*2+1]);  
}
inline int smax(int i,int j,int nod){//线段树找最大
    if(lt[nod]>=i&&rt[nod]<=j)return mt[nod];
    int mid=lt[nod]+rt[nod]>>1,ans=-1e9;
    if(i<=mid)ans=max(ans,smax(i,j,nod*2));
    if(j>mid)ans=max(ans,smax(i,j,nod*2+1));
    return ans;
}
inline int fmax(int x,int y){//树链上找最大
    int fx=top[x],fy=top[y],ans=-1e9;
    while(fx!=fy){
        if(deep[fx]<deep[fy])swap(fx,fy),swap(x,y);
        ans=max(ans,smax(sx[fx],sx[x],1));
        x=fa[fx];fx=top[x];
    }
    if(deep[x]>deep[y])swap(x,y);
    ans=max(ans,smax(sx[x],sx[y],1));
    return ans;
}
inline int ssum(int i,int j,int nod){//线段树求和
    if(lt[nod]>=i&&rt[nod]<=j)return t[nod];
    int mid=lt[nod]+rt[nod]>>1,ans=0;
    if(i<=mid)ans+=ssum(i,j,nod*2);
    if(j>mid)ans+=ssum(i,j,nod*2+1);
    return ans;
}
inline int fsum(int x,int y){//树链上求和
    int fx=top[x],fy=top[y],ans=0;
    while(fx!=fy){
        if(deep[fx]<deep[fy])swap(fx,fy),swap(x,y);
        ans+=ssum(sx[fx],sx[x],1);
        x=fa[fx];fx=top[x];
    }
    if(deep[x]>deep[y])swap(x,y);
    ans+=ssum(sx[x],sx[y],1);
    return ans;
}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<n;i++){
        int x,y;scanf("%d%d",&x,&y);
        addedge(x,y);addedge(y,x);
    }
    for(int i=1;i<=n;i++)scanf("%d",&v[i]);
    dfs(1,0,1);dfss(1,1);build(1,n,1);
    int m;scanf("%d",&m);
    for(int i=1;i<=m;i++){
        int x,y;scanf("%s%d%d",c+1,&x,&y);
        if(c[1]=='C')xg(sx[x],1,y);
        if(c[2]=='M')printf("%d\n",fmax(x,y));
        if(c[2]=='S')printf("%d\n",fsum(x,y));
    }
    return 0;
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值