HYSBZ4034-T2

4034: [HAOI2015]树上操作

Time Limit: 10 Sec   Memory Limit: 256 MB
Submit: 5552   Solved: 1783
[ Submit][ Status][ Discuss]

Description

有一棵点数为 N 的树，以点 1 为根，且树点有边权。然后有 M 个

操作，分为三种：

操作 1 ：把某个节点 x 的点权增加 a 。

操作 2 ：把某个节点 x 为根的子树中所有点的点权都增加 a 。

操作 3 ：询问某个节点 x 到根的路径中所有点的点权和。

Input

第一行包含两个整数 N, M 。表示点数和操作数。接下来一行 N 个整数，表示树中节点的初始权值。接下来 N-1 

行每行三个正整数 fr, to ， 表示该树中存在一条边 (fr, to) 。再接下来 M 行，每行分别表示一次操作。其中

第一个数表示该操作的种类（ 1-3 ） ，之后接这个操作的参数（ x 或者 x a ） 。

Output

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

Sample Input

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

Sample Output

6
9
13

HINT

 对于 100% 的数据， N,M<=100000 ，且所有输入数据的绝对值都不会超过 10^6 。

Source

鸣谢bhiaibogf提供

解题思路：树链剖分

#include <iostream>
#include <cstdio>
#include <cstring>
#include <string>
#include <algorithm>
#include <queue>
#include <stack>
#include <cmath>
#include <map>
#include <bitset>
#include <set>
#include <vector>
#include <functional>

using namespace std;

#define LL long long
const int INF = 0x3f3f3f3f;

int n,m,x,y,z;
int s[200009],nt[200009],e[200009],cnt;
int ct[200009],mx[200009],fa[200009],dep[200009];
int top[200009],L[200009],R[200009],g[200009],a[200009];
LL sum[200009<<2],lazy[200009<<2];

void dfs(int x,int f)
{
    dep[x]=dep[f]+1;
    fa[x]=f;ct[x]=1,mx[x]=0;
    for(int i=s[x];~i;i=nt[i])
    {
        if(e[i]==f) continue;
        dfs(e[i],x);
        ct[x]+=ct[e[i]];
        if(ct[e[i]]>ct[mx[x]]) mx[x]=e[i];
    }
}

void Dfs(int x,int t)
{
    top[x]=!t?x:top[fa[x]];
    L[x]=++cnt;g[cnt]=x;
    if(mx[x]) Dfs(mx[x],1);
    for(int i=s[x];~i;i=nt[i])
    {
        if(e[i]==fa[x]||e[i]==mx[x]) continue;
        Dfs(e[i],0);
    }
    R[x]=cnt;
}

void build(int k,int l,int r)
{
    lazy[k]=0;
    if(l==r) {sum[k]=a[g[l]];return ;}
    int mid=(l+r)>>1;
    build(k<<1,l,mid);
    build(k<<1|1,mid+1,r);
    sum[k]=sum[k<<1]+sum[k<<1|1];
}

LL query(int k,int l,int r,int ll,int rr)
{
    if(l>=ll&&r<=rr) return sum[k];
    int mid=(l+r)>>1;
    if(lazy[k])
    {
        lazy[k<<1]+=lazy[k],lazy[k<<1|1]+=lazy[k];
        sum[k<<1]+=lazy[k]*(mid-l+1),sum[k<<1|1]+=lazy[k]*(r-mid);
        lazy[k]=0;
    }
    LL ans=0;
    if(ll<=mid) ans+=query(k<<1,l,mid,ll,rr);
    if(rr>mid) ans+=query(k<<1|1,mid+1,r,ll,rr);
    return ans;
}

void update(int k,int l,int r,int ll,int rr,int val)
{
    if(l>=ll&&r<=rr) {sum[k]+=1LL*val*(r-l+1);lazy[k]+=val;return ;}
    int mid=(l+r)>>1;
    if(lazy[k])
    {
        lazy[k<<1]+=lazy[k],lazy[k<<1|1]+=lazy[k];
        sum[k<<1]+=lazy[k]*(mid-l+1),sum[k<<1|1]+=lazy[k]*(r-mid);
        lazy[k]=0;
    }
    if(ll<=mid) update(k<<1,l,mid,ll,rr,val);
    if(rr>mid) update(k<<1|1,mid+1,r,ll,rr,val);
    sum[k]=sum[k<<1]+sum[k<<1|1];
}

LL getans(int x,int y)
{
    LL ans=0;
    while(top[x]!=top[y])
    {
        if(dep[top[x]]<dep[top[y]]) swap(x,y);
        ans+=query(1,1,n,L[top[x]],L[x]);x=fa[top[x]];
    }
    if(dep[x]>dep[y]) swap(x,y);
    return ans+query(1,1,n,L[x],L[y]);
}

int main()
{
    while(~scanf("%d%d",&n,&m))
    {
        memset(s,-1,sizeof s);
        ct[0]=dep[0]=cnt=0;
        for(int i=1;i<=n;i++) scanf("%d",&a[i]);
        for(int i=1;i<n;i++)
        {
            scanf("%d%d",&x,&y);
            nt[cnt]=s[x],s[x]=cnt,e[cnt++]=y;
            nt[cnt]=s[y],s[y]=cnt,e[cnt++]=x;
        }
        dfs(1,0);
        Dfs(1,cnt=0);
        build(1,1,n);
        while(m--)
        {
            scanf("%d",&x);
            if(x==3)
            {
                scanf("%d",&y);
                printf("%lld\n",getans(1,y));
            }
            else
            {
                scanf("%d%d",&y,&z);
                update(1,1,n,L[y],x==1?L[y]:R[y],z);
            }
        }
    }
    return 0;
}