2019 ICPC 上海 F 题 A Simple Problem On A Tree

题目链接

题意

给出一棵 $n$ 个点的树，每个点有点权。给出 $q$ 个操作，有 $4$ 种操作：

• 把 $(u,v)$ 路径上所有点点权置为 $w$。
• 把 $(u,v)$ 路径上所有点点权加上 $w$。
• 把 $(u,v)$ 路径上所有点点权乘上 $w$。
• 求 $(u,v)$ 路径上所有点点权 ${\sum }_x{W}_x^3$ （立方和）。

题解

树链剖分把树上问题转化为序列上的问题，$3$ 修改种操作使用 $2$ 个标记就可以实现，都是 $\times x+y$ 的形式

• 第一种操作视为 $\times 0+w$
• 第二种操作视为 $\times 1+w$
• 第三种操作视为 $\times w+0$。

所以维护和，平方和，立方和，乘法标记和加法标记即可。

#include <bits/stdc++.h>
#define lc rt << 1
#define rc rt << 1 | 1
using namespace std;
const int N = 1e5 + 10;