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
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
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
