(color{#0066ff}{题目描述})
有一棵点数为 N 的树，以点 1 为根，且树点有边权。然后有 M 个操作，分为三种：
操作 1 ：把某个节点 x 的点权增加 a 。
操作 2 ：把某个节点 x 为根的子树中所有点的点权都增加 a 。
操作 3 ：询问某个节点 x 到根的路径中所有点的点权和。
(color{#0066ff}{输入格式})
第一行包含两个整数 N, M 。表示点数和操作数。
接下来一行 N 个整数，表示树中节点的初始权值。
接下来 N-1 行每行两个正整数 from, to ， 表示该树中存在一条边 (from, to) 。
再接下来 M 行，每行分别表示一次操作。其中第一个数表示该操作的种类（ 1-3 ） ，之后接这个操作的参数（ x 或者 x a ） 。
(color{#0066ff}{输出格式})
对于每个询问操作，输出该询问的答案。答案之间用换行隔开。
(color{#0066ff}{输入样例})
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
(color{#0066ff}{输出样例})
6
9
13
(color{#0066ff}{数据范围与提示})
对于 100% 的数据， N,M<=100000 ，且所有输入数据的绝对值都不会超过 10^6 。
(color{#0066ff}{题解})
操作1.2，线段树上直接修改
操作3，暴力跳top
不开LL见祖宗
#include <bits/stdc++.h>
#define LL long long
const int maxn = 1e5 + 10;
LL in() {
char ch; LL x = 0, f = 1;
while(!isdigit(ch = getchar()))(ch == '-') && (f = -f);
while(isdigit(ch)) x = (x << 1) + (x << 3) + (ch ^ 48), ch = getchar();
return x * f;
}
struct node {
int to;
node *nxt;
node(int to = 0, node *nxt = NULL):to(to), nxt(nxt) {}
void *operator new (size_t) {
static node *S = NULL, *T = NULL;
return (S == T) && (T = (S = new node[1024]) + 1024), S++;
}
};
struct sgt {
int l, r;
LL tag, val;
sgt *ch[2];
sgt(int l = 0, int r = 0, int tag = 0, int val = 0):l(l), r(r), tag(tag), val(val) {}
void *operator new (size_t) {
static sgt *S = NULL, *T = NULL;
return (S == T) && (T = (S = new sgt[1024]) + 1024), S++;
}
LL siz() { return r - l + 1; }
void upd() { val = ch[0]->val + ch[1]->val; }
void dwn() {
if(!tag) return;
ch[0]->val += ch[0]->siz() * tag;
ch[1]->val += ch[1]->siz() * tag;
ch[0]->tag += tag;
ch[1]->tag += tag;
tag = 0;
}
};
int val[maxn], dep[maxn], top[maxn], dfn[maxn], redfn[maxn], fa[maxn], son[maxn], siz[maxn], cnt;
int n, m;
node *head[maxn];
sgt *root;
void add(int from, int to) {
node *o = new node(to, head[from]);
head[from] = o;
}
void dfs1(int x, int f) {
fa[x] = f;
siz[x] = 1;
dep[x] = dep[f] + 1;
for(node *i = head[x]; i; i = i->nxt) {
if(i->to == f) continue;
dfs1(i->to, x);
siz[x] += siz[i->to];
if(!son[x] || siz[i->to] > siz[son[x]]) son[x] = i->to;
}
}
void dfs2(int x, int t) {
top[x] = t;
dfn[x] = ++cnt;
redfn[cnt] = x;
if(son[x]) dfs2(son[x], t);
for(node *i = head[x]; i; i = i->nxt)
if(!dfn[i->to]) dfs2(i->to, i->to);
}
void build(sgt *&o, int l, int r) {
o = new sgt(l, r, 0, 0);
if(l == r) return (void)(o->val = val[redfn[l]]);
int mid = (l + r) >> 1;
build(o->ch[0], l, mid);
build(o->ch[1], mid + 1, r);
o->upd();
}
void lazy(sgt *o, int l, int r, int k) {
if(o->r < l || o->l > r) return;
if(l <= o->l && o->r <= r) {
o->tag += k;
o->val += o->siz() * k;
return;
}
o->dwn();
lazy(o->ch[0], l, r, k), lazy(o->ch[1], l, r, k);
o->upd();
}
LL query(sgt *o, int l, int r) {
if(o->r < l || o->l > r) return 0;
if(l <= o->l && o->r <= r) return o->val;
o->dwn();
return query(o->ch[0], l, r) + query(o->ch[1], l, r);
}
LL querypath(int x) {
LL ans = 0;
int fx = top[x];
while(x) {
fx = top[x];
ans += query(root, dfn[fx], dfn[x]);
x = fa[fx];
}
return ans;
}
int main() {
n = in(), m = in();
int p, x;
for(int i = 1; i <= n; i++) val[i] = in();
for(int i = 1; i < n; i++) p = in(), x = in(), add(p, x), add(x, p);
dfs1(1, 0);
dfs2(1, 1);
build(root, 1, n);
while(m--) {
p = in(), x = in();
if(p == 1) lazy(root, dfn[x], dfn[x], in());
if(p == 2) lazy(root, dfn[x], dfn[x] + siz[x] - 1, in());
if(p == 3) printf("%lld
", querypath(x));
}
return 0;
}