洛谷P3178 [HAOI2015]树上操作
标签
- 树链剖分
简明题意
- 无
思路
- 裸的树剖,秒了啊啊啊啊啊!!!
注意事项
- 好吧,其实也没能一次过。总结下来,就是支持修改的数据结构,你得看他修改次数*修改范围会不会爆long long。然后,如果爆了long long,除了修改变量成long long,不要忘记返回值也要改成long long…
总结
- 如上
AC代码
#include<cstdio>
#include<vector>
#include<algorithm>
using namespace std;
const int maxn = 1e5 + 10;
int n, m, a[maxn];
vector<int> g[maxn];
int dep[maxn], fa[maxn], siz[maxn], son[maxn];
void dfs1(int u, int f, int deep)
{
dep[u] = deep;
fa[u] = f;
siz[u] = 1;
int maxson = -1;
for (auto& v : g[u])
if (v != f)
{
dfs1(v, u, deep + 1);
siz[u] += siz[v];
if (siz[v] > maxson)
maxson = siz[v], son[u] = v;
}
}
int top[maxn], id[maxn], cnt, w[maxn];
void dfs2(int u, int topf)
{
id[u] = ++cnt;
w[cnt] = a[u];
top[u] = topf;
if (son[u])
{
dfs2(son[u], topf);
for (auto& v : g[u])
if (v != fa[u] && v != son[u])
dfs2(v, v);
}
}
struct Node
{
int l, r;
long long tag, sum;
};
Node tree[maxn * 4];
void spread(int o)
{
if (tree[o].tag)
{
tree[o].sum += tree[o].tag * (tree[o].r - tree[o].l + 1);
if (tree[o].l != tree[o].r)
tree[o * 2].tag += tree[o].tag, tree[o * 2 + 1].tag += tree[o].tag;
tree[o].tag = 0;
}
}
void update(int o)
{
if (tree[o].l != tree[o].r)
{
spread(o * 2), spread(o * 2 + 1);
tree[o].sum = tree[o * 2].sum + tree[o * 2 + 1].sum;
}
}
void build(int o, int l, int r)
{
tree[o].l = l, tree[o].r = r;
if (l == r)
{
tree[o].sum = w[l];
return;
}
int mid = (l + r) / 2;
build(o * 2, l, mid);
build(o * 2 + 1, mid + 1, r);
update(o);
}
long long ask(int o, int l, int r)
{
spread(o);
if (tree[o].l == l && tree[o].r == r)
return tree[o].sum;
int mid = (tree[o].l + tree[o].r) / 2;
if (r <= mid)
return ask(o * 2, l, r);
else if (l > mid)
return ask(o * 2 + 1, l, r);
else
return ask(o * 2, l, mid) + ask(o * 2 + 1, mid + 1, r);
}
void change(int o, int l, int r, int c)
{
spread(o);
if (tree[o].l == l && tree[o].r == r)
{
tree[o].tag += c;
spread(o);
return;
}
int mid = (tree[o].l + tree[o].r) / 2;
if (r <= mid)
change(o * 2, l, r, c);
else if (l > mid)
change(o * 2 + 1, l, r, c);
else
change(o * 2, l, mid, c), change(o * 2 + 1, mid + 1, r, c);
update(o);
}
void solve()
{
scanf("%d%d", &n, &m);
for (int i = 1; i <= n; i++)
scanf("%d", &a[i]);
for (int i = 1; i < n; i++)
{
int u, v;
scanf("%d%d", &u, &v);
g[u].push_back(v), g[v].push_back(u);
}
dfs1(1, 1, 1);
dfs2(1, 1);
build(1, 1, n);
while (m--)
{
int opt;
scanf("%d", &opt);
if (opt == 1)
{
int x, c;
scanf("%d%d", &x, &c);
change(1, id[x], id[x], c);
}
else if (opt == 2)
{
int x, c;
scanf("%d%d", &x, &c);
change(1, id[x], id[x] + siz[x] - 1, c);
}
else if (opt == 3)
{
int u;
scanf("%d", &u);
int v = 1;
long long ans = 0;
while (top[u] != top[v])
{
if (dep[top[v]] < dep[top[u]]) swap(u, v);
ans += ask(1, id[top[v]], id[v]);
v = fa[top[v]];
}
ans += ask(1, min(id[u], id[v]), max(id[u], id[v]));
printf("%lld\n", ans);
}
}
}
int main()
{
freopen("Testin.txt", "r", stdin);
solve();
return 0;
}
双倍经验
- 无