【luogu P4074】【ybt金牌导航6-4-3】糖果公园（带修莫队）（树上莫队）
糖果公园
题目链接：luogu P4074 / ybt金牌导航6-4-3
题目大意
给你一棵树，树上点有颜色，颜色有权值，然后还给出新鲜度。
然后有一些操作，可能是修改一个点的颜色，可能是给你一条路径，问你路径的分数。
路径的分数是按顺序经过点，每个点给分数的贡献是它颜色的权值乘上新鲜度的第 i 项（i 是你第几次经到这个颜色的点）。
询问之间独立。
思路
这道题其实就是模板的带修树上莫队题。
首先就是如何分块，我们考虑用这样的一种分块方式：
我们每遍历处理完一个点的子树，就把这个点放入队列中。然后如果当前这个点的儿子要处理的点数超过了大小，就把那些找出来标成一类。
然后接着有时间就是 \(n^{\frac{2}{3}}\) 块长，然后就是正常的搞搞待修莫队。
然后你会发现询问的修改好像不知道要怎么移动？
然后其实你通过观察会发现，我们莫队不是有一个点是否贡献吗，你会发现你原来是 \((x1,y1)\)，现在是 \((x2,y2)\)，我们其实就把 \(x1\) 到 \(x2\) 和 \(y1\) 到 \(y2\) 这两个路径上的点状态取反即可。
由于我们是莫队，那我们就可以暴力的移动修改。
然后 ybt 要卡常，我卡了三页真的卡不动了。
代码（这个ybt会只有70）
//#pragma GCC optimize(2)
#include<cmath>
#include<cstdio>
#include<algorithm>
#define ll long long
using namespace std;
const int N = 100005;
struct ask {
int t, x, y, num;
}a[N];
int pl[N], bef[N], to[N], num[N];
int n, m, q, v[N], w[N], bn;
int tim, qn, op, x, y, t;
int c[N], lst[N], sz, bl[N], LCA;
int deg[N], fa[N][18], sta[N];
bool in[N];
ll ans, an[N];
struct node {
int to, nxt;
}e[N << 1];
int le[N], KK;
inline void add(int x, int y) {
e[++KK] = (node){y, le[x]}; le[x] = KK;
e[++KK] = (node){x, le[y]}; le[y] = KK;
}
inline int read() {
int re = 0;
char c = getchar();
while (c < '0' || c > '9') c = getchar();
while (c >= '0' && c <= '9') {
re = (re << 3) + (re << 1) + c - '0';
c = getchar();
}
return re;
}
inline int dfs_bl(int now, int father) {//分块
fa[now][0] = father;
deg[now] = deg[father] + 1;
int sn = 0;
for (int i = le[now]; i; i = e[i].nxt)
if (e[i].to ^ father) {
sn += dfs_bl(e[i].to, now);
if (sn >= sz) {
bn++;
while (sn) {
bl[sta[sta[0]]] = bn;
sta[0]--; sn--;
}
}
}
sta[++sta[0]] = now; sn++;
return sn;
}
bool cmp(ask x, ask y) {//莫队排序
if (bl[x.x] ^ bl[y.x]) return bl[x.x] < bl[y.x];
if (bl[x.y] ^ bl[y.y]) return (bl[x.x] & 1) ? bl[x.y] < bl[y.y] : bl[x.y] > bl[y.y];
return ((bl[x.x] ^ bl[x.y]) & 1) ? x.t < y.t : x.t > y.t;
}
inline int lca(int x, int y) {
if (deg[y] > deg[x]) swap(x, y);
for (int i = 17; i >= 0; i--)
if (deg[fa[x][i]] >= deg[y]) x = fa[x][i];
if (x == y) return x;
for (int i = 17; i >= 0; i--)
if (fa[x][i] ^ fa[y][i]) x = fa[x][i], y = fa[y][i];
return fa[x][0];
}
inline void putans(int x) {//更新答案（一个点）
if (!in[x]) {
num[c[x]]++;
ans += 1ll * v[c[x]] * w[num[c[x]]];
}
else {
ans -= 1ll * v[c[x]] * w[num[c[x]]];
num[c[x]]--;
}
in[x] ^= 1;
}
void change(int x, int y) {//给树上点更颜色
if (!in[x]) c[x] = y;
else {
putans(x);
c[x] = y;
putans(x);
}
}
inline void turn(int x, int y) {//给树上路径上的所有点统计（LCA不会统计）
while (x ^ y) {
if (deg[x] > deg[y]) putans(x), x = fa[x][0];
else putans(y), y = fa[y][0];
}
}
inline void write(ll x) {
if (x > 9ll) write(x / 10);
putchar(x % 10 + '0');
}
int main() {
n = read(); m = read(); q = read();
for (int i = 1; i <= m; i++) v[i] = read();
for (int i = 1; i <= n; i++) w[i] = read();
for (int i = 1; i < n; i++) {
x = read(); y = read();
add(x, y);
}
for (int i = 1; i <= n; i++) c[i] = read(), lst[i] = c[i];
for (int i = 1; i <= q; i++) {
op = read(); x = read(); y = read();
if (op == 0) {
tim++;
pl[tim] = x; bef[tim] = lst[x]; to[tim] = y;
lst[x] = y;
}
else {
a[++qn] = (ask){tim, x, y, qn};
}
}
sz = pow(n, 2.0 / 3);
int tmp = dfs_bl(1, 0); if (tmp) bn++;
while (tmp) {
bl[sta[sta[0]]] = bn;
sta[0]--; tmp--;
}
for (int i = 1; i <= 17; i++)
for (int j = 1; j <= n; j++)
fa[j][i] = fa[fa[j][i - 1]][i - 1];
sort(a + 1, a + qn + 1, cmp);
t = a[1].t;
for (int i = 1; i <= t; i++) change(pl[i], to[i]);
turn(a[1].x, a[1].y);
LCA = lca(a[1].x, a[1].y);
putans(LCA);
an[a[1].num] = ans;
putans(LCA);
for (int i = 2; i <= qn; i++) {//莫队
while (t < a[i].t) t++, change(pl[t], to[t]);
while (t > a[i].t) change(pl[t], bef[t]), t--;
turn(a[i].x, a[i - 1].x);
turn(a[i].y, a[i - 1].y);
LCA = lca(a[i].x, a[i].y);
putans(LCA);//前面没有处理到 LCA 所以要单独处理
an[a[i].num] = ans;
putans(LCA);
}
for (int i = 1; i <= qn; i++) write(an[i]), putchar('\n');
return 0;
}
后记
使用了新的树上莫队，用了欧拉序然后就过了。
新代码
#include<cmath>
#include<cstdio>
#include<algorithm>
#define ll long long
using namespace std;
const int N = 100001;
struct ask {
int t, x, y, num, lca;
}a[N];
int pl[N], bef[N], to[N], num[N];
int n, m, q, v[N], w[N], bn, tmpn;
int tim, qn, op, x, y, t, dfn[2 * N];
int c[N], lst[N], sz, bl[2 * N], LCA;
int deg[N], fa[N][18], st[N], ed[N];
bool in[N];
ll ans, an[N];
struct node {
int to, nxt;
}e[2 * N];
int le[N], KK;
inline void add(int x, int y) {
e[++KK] = (node){y, le[x]}; le[x] = KK;
e[++KK] = (node){x, le[y]}; le[y] = KK;
}
int re;
char cc;
inline int read() {
re = 0;
cc = getchar();
while (cc < '0' || cc > '9') cc = getchar();
while (cc >= '0' && cc <= '9') {
re = (re << 3) + (re << 1) + cc - '0';
cc = getchar();
}
return re;
}
inline void dfs_bl(int now, int father) {
dfn[++tmpn] = now;
st[now] = tmpn;
fa[now][0] = father;
deg[now] = deg[father] + 1;
for (int i = le[now]; i; i = e[i].nxt)
if (e[i].to ^ father) {
dfs_bl(e[i].to, now);
}
dfn[++tmpn] = now;
ed[now] = tmpn;
}
inline bool cmp(ask x, ask y) {
// if (bl[x.x] ^ bl[y.x]) return bl[x.x] < bl[y.x];
// if (bl[x.y] ^ bl[y.y]) return (bl[x.x] & 1) ? bl[x.y] < bl[y.y] : bl[x.y] > bl[y.y];
// return ((bl[x.x] ^ bl[x.y]) & 1) ? x.t < y.t : x.t > y.t;
if (bl[x.x] ^ bl[y.x]) return bl[x.x] < bl[y.x];
if (bl[x.y] ^ bl[y.y]) return bl[x.y] < bl[y.y];
return x.t < y.t;
}
inline int lca(int x, int y) {
if (deg[y] > deg[x]) swap(x, y);
for (int i = 17; i >= 0; i--)
if (deg[fa[x][i]] >= deg[y]) x = fa[x][i];
if (x == y) return x;
for (int i = 17; i >= 0; i--)
if (fa[x][i] ^ fa[y][i]) x = fa[x][i], y = fa[y][i];
return fa[x][0];
}
inline void putans(int x) {
if (!in[x]) {
num[c[x]]++;
ans += 1ll * v[c[x]] * w[num[c[x]]];
}
else {
ans -= 1ll * v[c[x]] * w[num[c[x]]];
num[c[x]]--;
}
in[x] ^= 1;
}
inline void change(int x, int y) {
if (!in[x]) c[x] = y;
else {
putans(x);
c[x] = y;
putans(x);
}
}
void write(ll x) {
if (x > 9ll) write(x / 10);
putchar(x % 10 + '0');
}
int main() {
n = read(); m = read(); q = read();
for (int i = 1; i <= m; i++) v[i] = read();
for (int i = 1; i <= n; i++) w[i] = read();
for (int i = 1; i < n; i++) {
x = read(); y = read();
add(x, y);
}
for (int i = 1; i <= n; i++) c[i] = read(), lst[i] = c[i];
sz = pow(2 * n, 2.0 / 3);
dfs_bl(1, 0);
for (int i = 1; i <= 17; i++)
for (int j = 1; j <= n; j++)
fa[j][i] = fa[fa[j][i - 1]][i - 1];
for (int i = 1; i <= 2 * n; i++) bl[i] = (i - 1) / sz + 1;
for (int i = 1; i <= q; i++) {
op = read(); x = read(); y = read();
if (op == 0) {
tim++;
pl[tim] = x; bef[tim] = lst[x]; to[tim] = y;
lst[x] = y;
}
else {
LCA = lca(x, y);
if (st[x] > st[y]) swap(x, y);
if (LCA == x) a[++qn] = (ask){tim, st[x], st[y], qn, 0};
else a[++qn] = (ask){tim, ed[x], st[y], qn, LCA};
}
}
sort(a + 1, a + qn + 1, cmp);
x = 1; y = 0; t = 0;
for (int i = 1; i <= qn; i++) {
while (t < a[i].t) t++, change(pl[t], to[t]);
while (t > a[i].t) change(pl[t], bef[t]), t--;
while (x < a[i].x) putans(dfn[x]), x++;
while (x > a[i].x) x--, putans(dfn[x]);
while (y < a[i].y) y++, putans(dfn[y]);
while (y > a[i].y) putans(dfn[y]), y--;
if (a[i].lca) putans(a[i].lca);
an[a[i].num] = ans;
if (a[i].lca) putans(a[i].lca);
}
for (int i = 1; i <= qn; i++) write(an[i]), putchar('\n');
return 0;
}
