BZOJ4034: [HAOI2015]树上操作

Posted 2020-10-26 mt-li

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 。

 

题目传送门

鸣谢AKCqhzdy的指点（以后他应该是我博客常驻居民了）

终于学会了树状数组+差分了。。。。

这东西真tm强悍

AKCqhzdy大佬博客：嘻嘻嘻我是大佬

 

代码如下：

#include<cmath>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
using namespace std;
typedef long long ll;
struct node{
    int x,y,next;
}a[210000];int len,last[210000];
void ins(int x,int y)
{
    len++;
    a[len].x=x;a[len].y=y;
    a[len].next=last[x];last[x]=len;
}
int n,m,dep[210000];
int l[210000],r[210000],id;
void pre_tree_node(int x,int fa)
{
    l[x]=++id;
    for(int k=last[x];k;k=a[k].next)
    {
        int y=a[k].y;
            if(y!=fa)
                dep[y]=dep[x]+1,pre_tree_node(y,x);
    }
    r[x]=id;
}
ll c[2][210000];
int lowbit(int x){return x&-x;}
void change(int w,int x,ll k)
{
    while(x<=n)
    {
        c[w][x]+=k;
        x+=lowbit(x);
    }
}
ll getsum(int w,int x)
{
    ll sum=0;
    while(x)
    {
        sum+=c[w][x];
        x-=lowbit(x);
    }
    return sum;
}
ll d[210000];
int main()
{
    scanf("%d%d",&n,&m);
    len=0;memset(last,0,sizeof(last));
    for(int i=1;i<=n;i++)scanf("%lld",&d[i]);
    
    for(int i=1;i<n;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        ins(x,y),ins(y,x);
    }
    id=0,dep[1]=1,pre_tree_node(1,0);
    for(int i=1;i<=n;i++)
        change(0,l[i],d[i]),change(0,r[i]+1,-d[i]);
    int op;
    while(m--)
    {
        scanf("%d",&op);
        if(op==1)
        {
            int x;ll p;
            scanf("%d%lld",&x,&p);
            change(0,l[x],p);change(0,r[x]+1,-p);
        }
        else if(op==2)
        {
            int x;ll p;
            scanf("%d%lld",&x,&p);
            change(0,l[x],-p*(dep[x]-1)),change(0,r[x]+1,p*(dep[x]-1));
            change(1,l[x],p),change(1,r[x]+1,-p);
        }
        else
        {
            int x;
            scanf("%d",&x);
            printf("%lld\\n",getsum(0,l[x])+dep[x]*getsum(1,l[x]));
        }
    }
    return 0;
} 

by_lmy