树链剖分习题

最新推荐文章于 2022-04-01 14:24:51 发布

原创最新推荐文章于 2022-04-01 14:24:51 发布 · 340 阅读

1 ·

CC 4.0 BY-SA版权

树链剖分专栏收录该内容

6 篇文章

订阅专栏

（先记录一下有几个好题，以后会慢慢做完）
1.洛谷 P2486 染色
2.洛谷 P3187 树上操作
一开始中间查询值的函数query中ans定义的是int类型，然后就错了，只对了5个测试点，后面改为longlong之后才对。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#define ls (rt<<1)
#define rs (rt<<1|1)
#define mid (tree[rt].l+(tree[rt].r-tree[rt].l)/2)
using namespace std;
struct arr{
    long long l,r,sum,tag;
}tree[500000];
int w[200000],id[200000],deep[200000],fa[200000],sz[200000];
int top[200000],son[200000],pos[200000];
int n,m,rr,cnt;
vector<int>g[2000000];
template<typename tp>void read(tp & dig){//读入优化 
    char c=getchar();dig=0;
    while(!isdigit(c))c=getchar();
    while(isdigit(c))dig=dig*10+c-'0',c=getchar();
}
inline int read(){
    int x=0,w=1;char ch=0;
    while(ch!='-'&&(ch<'0'||ch>'9')) ch=getchar();
    if(ch=='-') w=-1,ch=getchar();
    while(ch>='0'&&ch<='9')x=x*10+ch-48,ch=getchar();
    return x*w;
}
inline void pushup(int rt){tree[rt].sum=tree[ls].sum+tree[rs].sum;}
inline void build(int l,int r,int rt){//建立一棵线段树 
    tree[rt].l=l,tree[rt].r=r;
    if(l==r){tree[rt].sum=w[id[l]];return ;}
    int midd=l+(r-l)/2;
    build(l,midd,ls),build(midd+1,r,rs);
    pushup(rt);
}
inline void pushdown(int rt){//下传标记 
    if(tree[rt].tag){
        tree[ls].tag=(tree[ls].tag+tree[rt].tag);
        tree[rs].tag=(tree[rs].tag+tree[rt].tag);
        tree[ls].sum=(tree[ls].sum+(tree[ls].r-tree[ls].l+1)*tree[rt].tag);
        tree[rs].sum=(tree[rs].sum+(tree[rs].r-tree[rs].l+1)*tree[rt].tag);
        tree[rt].tag=0;
    }
}
void update(int l,int r,int c,int rt){//区间、单点修改 
    if(l<=tree[rt].l&&tree[rt].r<=r){   //线段树的另一种写法，快一些 
        tree[rt].sum=(tree[rt].sum+(c*(tree[rt].r-tree[rt].l+1)));
        tree[rt].tag+=c;
        return ;
    }
    pushdown(rt);
    if(l<=mid) update(l,r,c,ls);
    if(r>mid) update(l,r,c,rs);
    pushup(rt);
}
long long query(int l,int r,int rt){//区间查询 
    if(l<=tree[rt].l&&tree[rt].r<=r) return tree[rt].sum;
    pushdown(rt);
    long long ans=0;
    if(l<=mid) ans+=query(l,r,ls);//？ 
    if(r>mid) ans+=query(l,r,rs);
    return ans;
}
void dfs1(int x,int fat,int dep){
    deep[x]=dep,fa[x]=fat,sz[x]=1;
    for(int i=0;i<g[x].size();i++){
        int v=g[x][i];
        if(v!=fat){
            dfs1(v,x,dep+1);
            sz[x]+=sz[v];
            if(son[x]==-1||sz[son[x]]<sz[v]) son[x]=v;
        }
    }
}
void dfs2(int x,int tp){
    top[x]=tp;pos[x]=++cnt;id[pos[x]]=x;
    if(son[x]==-1) return;
    dfs2(son[x],tp);
    for(int i=0;i<g[x].size();i++){
        int v=g[x][i];
        if(v!=fa[x]&&v!=son[x]) dfs2(v,v);
    }
}
long long add(int t1,int t2,int data,int ok){
    long long u=t1,v=t2,ans=0;
    while(top[u]!=top[v]){
        if(deep[top[u]]>deep[top[v]]) swap(u,v);
        if(!ok)update(pos[top[v]],pos[v],data,1);
        else ans+=query(pos[top[v]],pos[v],1);
        v=fa[top[v]];
    }
    if(deep[u]>deep[v]) swap(u,v);
    if(!ok) update(pos[u],pos[v],data,1);
    else ans+=query(pos[u],pos[v],1);
    return ans;
}
int main(){
    memset(son,-1,sizeof(son));
    n=read();m=read();
    for(int i=1;i<=n;i++) w[i]=read();
    for(int i=1;i<=n-1;i++){
        int u=read(),v=read(); g[u].push_back(v);g[v].push_back(u);
    }
    dfs1(1,-1,1);
    dfs2(1,1);
    build(1,n,1);//建树 
    for(int i=1;i<=m;i++){
        int xx,t1,t2,t3;
        cin>>xx;
        if(xx==1) t1=read(),t2=read(),add(t1,t1,t2,0); 
        if(xx==2) t1=read(),t2=read(),update(pos[t1],pos[t1]+sz[t1]-1,t2,1);
        if(xx==3) t1=read(),printf("%lld\n",add(t1,1,0,1));
    }
    return 0;
}