\(\mathcal{Description}\)
Link.
给定一棵 \(n\) 个点的带点权树,删除 \(u\) 点的代价是该点点权 \(a_u\)。\(m\) 次操作:
- 修改单点点权。
- 询问让某棵子树的根不可到达子树内任意一片叶子的代价。
\(n,m\le2\times10^5\)。
\(\mathcal{Solution}\)
不考虑修改,列出 DP:
\[f(u)=\begin{cases}a_u&u\text{ is leaf}\\\min\{a_u,\sum_vf(v)\}&\text{otherwise}\end{cases}
\]
单独拿出实儿子 \(s_u\):
\[g(u)=\begin{cases}+\infty&u\text{ is leaf}\\\sum_{v\not=s_u}f(v)&\text{otherwise}\end{cases}\\\Rightarrow f(u)=\min\{a_u,f(s_u)+g(u)\}
\]
定义矩乘的 \(+\) 为加法,\(\times\) 为取 \(\min\),有:
\[\begin{bmatrix}f(u)\\0\end{bmatrix}=\begin{bmatrix}g(u)&a_u\\+\infty&0\end{bmatrix}\begin{bmatrix}f(s_u)\\0\end{bmatrix}
\]
用 LCT / 树剖维护。若使用 LCT,询问 \(u\) 子树时,应 \(\operatorname{access}\) 原树上 \(u\) 的父亲,再 \(\operatorname{splay}\) \(u\),就能保证当前 \(u\) 的实链全部在子树内,输出 \(u\) 维护的矩乘答案即可。
还有一点,虽然 DP 是自下而上的,但把矩乘展开却是由 \(u\) 向实儿子走的。注意乘法顺序。
\(\mathcal{Code}\)
询问的时候 \(\operatorname{access}\) 成了 LCT 上的父亲调哭了 qwq。
#include <cstdio>
#define calc( x ) ( min_ ( S[x][0][0], S[x][0][1] ) )
typedef long long LL;
const int MAXN = 2e5;
const LL INF = 1e14;
int n, m, ecnt, a[MAXN + 5], head[MAXN + 5];
int srcfa[MAXN + 5], fa[MAXN + 5], ch[MAXN + 5][2];
inline LL min_ ( const LL a, const LL b ) { return a < b ? a : b; }
inline char fgc () {
static char buf[1 << 17], *p = buf, *q = buf;
return p == q && ( q = buf + fread ( p = buf, 1, 1 << 17, stdin ), p == q ) ? EOF : *p ++;
}
inline int rint () {
int x = 0, f = 1; char s = fgc ();
for ( ; s < '0' || '9' < s; s = fgc () ) f = s == '-' ? -f : f;
for ( ; '0' <= s && s <= '9'; s = fgc () ) x = x * 10 + ( s ^ '0' );
return x * f;
}
inline void wint ( LL x ) {
if ( x < 0 ) putchar ( '-' ), x = -x;
if ( 9 < x ) wint ( x / 10 );
putchar ( x % 10 ^ '0' );
}
struct Edge { int to, nxt; } graph[MAXN * 2 + 5];
struct Matrix {
LL mat[2][2];
Matrix (): mat { INF, INF, INF, INF } {}
inline LL* operator [] ( const int key ) { return mat[key]; }
inline Matrix operator * ( Matrix& t ) {
Matrix ret;
for ( int i = 0; i < 2; ++ i ) {
for ( int k = 0; k < 2; ++ k ) {
for ( int j = 0; j < 2; ++ j ) {
ret[i][j] = min_ ( ret[i][j], mat[i][k] + t[k][j] );
}
}
}
return ret;
}
} G[MAXN + 5], S[MAXN + 5];
inline void link ( const int s, const int t ) {
graph[++ ecnt] = { t, head[s] };
head[s] = ecnt;
}
inline bool nroot ( const int x ) { return ch[fa[x]][0] == x || ch[fa[x]][1] == x; }
inline void pushup ( const int x ) {
S[x] = G[x];
if ( ch[x][0] ) S[x] = S[ch[x][0]] * S[x];
if ( ch[x][1] ) S[x] = S[x] * S[ch[x][1]];
}
inline void rotate ( const int x ) {
int y = fa[x], z = fa[y], k = ch[y][1] == x;
fa[x] = z; if ( nroot ( y ) ) ch[z][ch[z][1] == y] = x;
ch[y][k] = ch[x][k ^ 1]; if ( ch[x][k ^ 1] ) fa[ch[x][k ^ 1]] = y;
pushup ( ch[fa[y] = x][k ^ 1] = y ), pushup ( x );
}
inline void splay ( const int x ) {
for ( int y; nroot ( x ); rotate ( x ) ) {
if ( nroot ( y = fa[x] ) ) {
rotate ( x ^ y ^ ch[y][0] ^ ch[fa[y]][0] ? x : y );
}
}
pushup ( x );
}
inline void access ( int x ) {
for ( int y = 0; x; x = fa[y = x] ) {
splay ( x );
if ( ch[x][1] ) G[x][0][0] += calc ( ch[x][1] );
if ( y ) G[x][0][0] -= calc ( y );
ch[x][1] = y, pushup ( x );
}
}
inline LL initDP ( const int u, const int fath ) {
LL sum = INF; fa[u] = srcfa[u] = fath;
for ( int i = head[u], v; i; i = graph[i].nxt ) {
if ( ( v = graph[i].to ) ^ fath ) {
if ( sum == INF ) sum = 0;
sum += initDP ( v, u );
}
}
G[u][0][0] = sum, G[u][0][1] = a[u], G[u][1][1] = 0;
return S[u] = G[u], min_ ( sum, a[u] );
}
int main () {
n = rint ();
for ( int i = 1; i <= n; ++ i ) a[i] = rint ();
for ( int i = 1, u, v; i < n; ++ i ) {
u = rint (), v = rint ();
link ( u, v ), link ( v, u );
}
initDP ( 1, 0 );
m = rint ();
for ( int x, t, op; m --; ) {
for ( op = fgc (); op < 'A' || 'Z' < op; op = fgc () );
x = rint ();
if ( op == 'Q' ) {
if ( srcfa[x] ) access ( srcfa[x] );
splay ( x );
wint ( calc ( x ) ), putchar ( '\n' );
} else {
t = rint ();
access ( x ), splay ( x );
G[x][0][1] += t, pushup ( x );
}
}
return 0;
}