2.4 【树链剖分】 HAOI2015 树上操作

HAOI2015 树上操作

题目描述

有一棵点数为 N 的树，以点 1 为根，且树点有边权。然后有 M 个操作，分为三种：
操作 1 ：把某个节点 x 的点权增加 a 。
操作 2 ：把某个节点 x 为根的子树中所有点的点权都增加 a 。
操作 3 ：询问某个节点 x 到根的路径中所有点的点权和。

输入格式

第一行包含两个整数 N, M 。表示点数和操作数。接下来一行 N 个整数，表示树中节点的初始权值。接下来 N-1 行每行两个正整数 from, to ， 表示该树中存在一条边 (from, to) 。再接下来 M 行，每行分别表示一次操作。其中第一个数表示该操作的种类（ 1-3 ） ，之后接这个操作的参数（ x 或者 x a ） 。

输出格式

对于每个询问操作，输出该询问的答案。答案之间用换行隔开。

输入样例

5 5
1 2 3 4 5
1 2
1 4
2 3
2 5
3 3
1 2 1
3 5
2 1 2
3 3

输出样例

6
9
13

树链剖分裸题，代码如下。

#include<cstdio>
#include<queue>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<vector>
#include<iostream>
#include<cstdlib>
using namespace std;
inline int read(){
    char c;int ret=0;bool flag=0;
    do{c=getchar();}while(c!='-'&&(c>'9'||c<'0'));
    if(c=='-'){c=getchar();flag=1;}
    do{ret=ret*10+c-48;c=getchar();}while(c>='0'&&c<='9');
    if(flag)return -ret;
    return ret;
}
int n,m;
struct E{
    int v,next;
};
E edge[300000];
int fir[100010];
int w[100010];
int summ=0;
int fa[100010];
int sons[100010],hson[100010];
vector<int> ha[100010];
int top[100010],pos[100010],cnt=0;
void add(int u,int v){
    edge[++summ].next=fir[u];
    fir[u]=summ;
    edge[summ].v=v;
}
void build(int x,int p){
    int bigone=0;
    int big=0;
    fa[x]=p;
    sons[x]=1;
    for(int i=fir[x];i;i=edge[i].next)
    {
        int v=edge[i].v;
        if(v==p)continue;
        build(v,x);
        sons[x]+=sons[v];
        ha[x].push_back(v);
        if(sons[v]>big)
        {
            big=sons[v];
            bigone=v;
        }
    }
    hson[x]=bigone;
}
void dfs(int x){
    pos[x]=++cnt;
    if(sons[x]==1)return;
    top[hson[x]]=top[x];
    dfs(hson[x]);
    for(int i=0;i<ha[x].size();i++)
    {
        int v=ha[x][i];
        if(v!=hson[x])
        {
            top[v]=v;
            dfs(v);
        }
    }
    return ;
}
long long sum[500000],v[500000];
void push(int now,int l,int r){
    if(l==r)return ;
    long long t=v[now];
    v[now]=0;
    int mid=(l+r)/2;
    v[now*2]+=t;
    v[now*2+1]+=t;
    sum[now*2]+=t*(mid-l+1);
    sum[now*2+1]+=t*(r-mid);
}
void addd(int now,int l,int r,int x,int y,long long val){
    if(v[now])push(now,l,r);    
    if(l==x&&r==y)
    {
        v[now]+=val;
        sum[now]+=(r-l+1)*val;
        return ;
    }
    int mid=(l+r)/2;
    if(x<=mid)
        addd(now*2,l,mid,x,min(mid,y),val);
    if(y>mid)
        addd(now*2+1,mid+1,r,max(x,mid+1),y,val);
    sum[now]=sum[now*2]+sum[now*2+1];
}
long long query(int now,int l,int r,int x,int y){
    if(v[now])push(now,l,r);
    if(l==x&&r==y)return (long long)sum[now];
    long long ret=0;
    int mid=(l+r)/2;
    if(x<=mid)
        ret+=query(now*2,l,mid,x,min(y,mid));
    if(y>mid)
        ret+=query(now*2+1,mid+1,r,max(x,mid+1),y);
    return ret;
}
int main(){
    n=read();m=read();
    int nn=n;
    for(int i=1;i<=n;i++)
        w[i]=read();
    int x,y;
    for(int i=1;i<n;i++)
    {
        x=read();y=read();
        add(x,y);
        add(y,x);
    }
    top[1]=1;
    build(1,0);
    dfs(1);
    for(int i=1;i<=n;i++)
        addd(1,1,n,pos[i],pos[i],w[i]);
    int cc;
    while(m--)
    {
        cc=read();
        if(cc==1)
        {
            x=read();y=read();
            addd(1,1,n,pos[x],pos[x],y);
        }
        else if(cc==2)
        {
            x=read();y=read();
            addd(1,1,n,pos[x],pos[x]+sons[x]-1,y);
        }
        else
        {
            x=read();
            long long anss=0;
            while(x!=0)
            {
                anss+=query(1,1,n,pos[top[x]],pos[x]);
                x=fa[top[x]];
            }
            printf("%lld\n",anss);
        }
    }
    return 0;
} 

LLAP.