BZOJ 4034[HAOI2015]树上操作 树链剖分
Code:
#include <bits/stdc++.h>
#define setIO(s) freopen(s".in","r",stdin), freopen(s".out","w",stdout)
#define maxn 110000
#define ll long long
#define lson (x<<1)
#define rson ((x<<1)|1)
using namespace std;
int hd[maxn], to[maxn << 1], nex[maxn << 1], st[maxn], ed[maxn];
int fa[maxn], dep[maxn], dfn[maxn], ln[maxn], hson[maxn], siz[maxn], bot[maxn], top[maxn];
int edges, n , Q, tim;
ll val[maxn];
void add(int u, int v)
{
nex[++edges] = hd[u], hd[u] = edges, to[edges] = v;
}
void dfs1(int u, int ff)
{
fa[u] = ff, dep[u] = dep[ff] + 1, siz[u] = 1;
for(int i = hd[u]; i ; i = nex[i])
{
int v = to[i];
if(v == ff) continue;
dfs1(v, u);
siz[u] += siz[v];
if(siz[v] > siz[hson[u]]) hson[u] = v;
}
}
void dfs2(int u, int tp)
{
top[u] = tp, ln[++tim] = u, dfn[u] = tim, st[u] = tim;
if(hson[u])
dfs2(hson[u], tp), bot[u] = bot[hson[u]];
else
bot[u] = u;
for(int i = hd[u]; i ; i = nex[i])
{
int v = to[i];
if(v == fa[u] || v == hson[u]) continue;
dfs2(v, v);
}
ed[u] = tim;
}
struct Node
{
ll lazy, val, sumv;
}t[maxn << 3];
void mark(int l, int r, int x, ll d)
{
t[x].lazy += d;
ll tmp = r - l + 1;
t[x].sumv += 1ll*tmp*1ll*d;
}
void pushdown(int l, int r, int x)
{
if(!t[x].lazy) return;
int mid = (l + r) >> 1;
if(mid >= l) mark(l, mid, lson, t[x].lazy);
if(r > mid) mark(mid + 1, r, rson, t[x].lazy);
t[x].lazy = 0;
}
void build(int l, int r, int x)
{
if(l == r)
{
t[x].sumv = t[x].val = val[ln[l]];
return;
}
int mid = (l + r) >> 1;
build(l, mid, lson);
build(mid + 1, r, rson);
t[x].sumv = t[lson].sumv + t[rson].sumv;
}
void update(int l, int r, int x, int L, int R, ll d)
{
pushdown(l, r, x);
if(l >= L && r <= R)
{
mark(l, r, x, d);
return;
}
int mid = (l + r) >> 1;
if(L <= mid) update(l, mid, lson, L, R, d);
if(R > mid) update(mid + 1, r, rson, L, R, d);
t[x].sumv = t[lson].sumv + t[rson].sumv;
}
ll query(int l, int r, int x, int L, int R)
{
pushdown(l, r, x);
if(l >= L && r <= R) return t[x].sumv;
int mid = (l + r) >> 1;
ll tmp = 0;
if(L <= mid) tmp += query(l, mid, lson, L, R);
if(R > mid) tmp += query(mid + 1, r, rson, L, R);
return tmp;
}
void modify1(int x, ll d)
{
update(1, n, 1, dfn[x], dfn[x], d);
}
void modify2(int x, ll d)
{
update(1, n, 1, st[x], ed[x], d);
}
ll Query(int x)
{
ll tmp = 0;
while(x)
{
tmp += query(1, n, 1, dfn[top[x]], dfn[x]);
x = fa[top[x]];
}
return tmp;
}
int main()
{
// setIO("input");
scanf("%d%d",&n,&Q);
for(int i = 1;i <= n; ++i) scanf("%lld",&val[i]);
for(int i = 1, u , v; i < n; ++i)
{
scanf("%d%d",&u,&v), add(u, v), add(v, u);
}
dfs1(1, 0), dfs2(1, 1), build(1, n, 1);
while(Q--)
{
int opt, x;
ll a;
scanf("%d",&opt);
if(opt == 1)
{
scanf("%d%lld",&x,&a), modify1(x, a);
}
if(opt == 2)
{
scanf("%d%lld",&x,&a), modify2(x, a);
}
if(opt == 3)
{
scanf("%d",&x);
printf("%lld\n",Query(x));
}
}
return 0;
}