ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

[模板]树链剖分

2022-05-18 16:34:53  阅读:125  来源: 互联网

标签:剖分 int res top 树链 WR son id 模板


简易版树链剖分

#include<cstdio>
#include<cstring>
#include<string>
#include<iostream>
#define WR WinterRain
using namespace std;
const int WR=1001000,INF=2147483647;
struct SegmentTree{
    int l,r,val,sum;
}tree[WR];
struct Edge{
    int pre,to;
}edge[WR];
int n,q;
int a[WR];
int head[WR],tot;
int fa[WR],dpt[WR],sze[WR],son[WR];
//fa存父亲,dpt存深度,sze存子树大小,son存重儿子
int top[WR],id[WR],rnk[WR],cnt;
int read(){
    int s=0,n=1;
    char ch=getchar();
    while(ch>'9'||ch<'0'){
        if(ch=='-') n=-1;
        ch=getchar();
    }
    while(ch>='0'&&ch<='9'){
        s=(s<<1)+(s<<3)+ch-48;
        ch=getchar();
    }
    return s*n;
}
void pushup(int k){
    tree[k].sum=tree[k<<1].sum+tree[k<<1|1].sum;
    tree[k].val=max(tree[k<<1].val,tree[k<<1|1].val);
}
void build(int k,int l,int r){
    tree[k].l=l,tree[k].r=r;
    if(l==r){
        tree[k].val=tree[k].sum=a[rnk[l]];
        return;
    }
    int mid=(l+r)>>1;
    build(k<<1,l,mid);
    build(k<<1|1,mid+1,r);
    pushup(k);
}
void modify(int k,int pos,int v){
    if(tree[k].l==tree[k].r){
        tree[k].val=tree[k].sum=v;
        return;
    }
    int mid=(tree[k].l+tree[k].r)>>1;
    if(pos<=mid) modify(k<<1,pos,v);
    else modify(k<<1|1,pos,v);
    pushup(k);
}
int query_sum(int k,int l,int r){
    if(tree[k].l>=l&&tree[k].r<=r) return tree[k].sum;
    int mid=(tree[k].l+tree[k].r)>>1,res=0;
    if(l<=mid) res+=query_sum(k<<1,l,r);
    if(r>mid) res+=query_sum(k<<1|1,l,r);
    return res;
}
int query_max(int k,int l,int r){
    if(tree[k].l>=l&&tree[k].r<=r) return tree[k].val;
    int mid=(tree[k].l+tree[k].r)>>1,res=-INF;
    if(l<=mid) res=max(res,query_max(k<<1,l,r));
    if(r>mid) res=max(res,query_max(k<<1|1,l,r));
    return res;
}
void add(int u,int v){
    edge[++tot].pre=head[u];
    edge[tot].to=v;
    head[u]=tot;
}
void dfs1(int u,int root){
    fa[u]=root,dpt[u]=dpt[root]+1,sze[u]=1;
    for(int i=head[u];i;i=edge[i].pre){
        int v=edge[i].to;
        if(v==root) continue;
        dfs1(v,u);
        sze[u]+=sze[v];
        if(sze[v]>sze[son[u]]) son[u]=v;//找最大的子树做重儿子
    }
}
void dfs2(int u,int tp){//tp是目前重链的顶端
    top[u]=tp,id[u]=++cnt,rnk[cnt]=u;//id和rnk存储剖分时访问到的次序
    if(son[u]) dfs2(son[u],tp);//有重儿子优先走重儿子
    for(int i=head[u];i;i=edge[i].pre){
        int v=edge[i].to;
        if(v!=fa[u]&&v!=son[u]) dfs2(v,v);//如果不是重儿子那么这个重链的顶端是自己 
    }
}
int get_sum(int x,int y){
    int res=0;
    while(top[x]!=top[y]){//x,y不在一条链上
        if(dpt[top[x]]<dpt[top[y]]) swap(x,y);//跳到链顶深度更大的
        res+=query_sum(1,id[top[x]],id[x]);//区间查询链
        x=fa[top[x]];//跳到链顶
    }
    if(dpt[x]>dpt[y]) swap(x,y);//确定高低位置
    //此时x和y肯定在一条重链上
    res+=query_sum(1,id[x],id[y]);
    return res;
}
int get_max(int x,int y){
    int res=-INF;
    while(top[x]!=top[y]){
        if(dpt[top[x]]<dpt[top[y]]) swap(x,y);
        res=max(res,query_max(1,id[top[x]],id[x]));
        x=fa[top[x]];
    }
    if(dpt[x]>dpt[y]) swap(x,y);
    res=max(res,query_max(1,id[x],id[y]));
    return res;
}
int main(){
    n=read();
    for(int i=1;i<n;i++){
        int u=read(),v=read();
        add(u,v);add(v,u);
    }
    for(int i=1;i<=n;i++) a[i]=read();
    dfs1(1,0);
    dfs2(1,1);
    build(1,1,n);
    q=read();
    for(int i=1;i<=q;i++){
        char opt[10];
        cin>>opt;
        int x=read(),y=read();
        if(opt[1]=='H') modify(1,id[x],y);
        if(opt[1]=='S') printf("%d\n",get_sum(x,y));
        if(opt[1]=='M') printf("%d\n",get_max(x,y));
    }
    return 0;
}
View Code

 

标签:剖分,int,res,top,树链,WR,son,id,模板
来源: https://www.cnblogs.com/WintersRain/p/16285139.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有