铲雪问题
题目链接:ybtoj高效进阶 21273
题目大意
给你一棵树,点有点权。
然后你每次可以选择一段路径,把路径上的点全部边权减一。
但是不许保证路径上的点点权都不为 0。
然后问你最少要进行多少次操作才能把树上的所有点的点权都减成 0。
思路
考虑先暴力的一条边一条边的删,然后考虑合并一些删除。
那就搞树形 DP。
f
i
f_i
fi 表示搞定
i
i
i 的子树能少删多少次,
g
i
g_i
gi 表示搞定完之后还有多少个能继续向上跟上面的点配对。
那不难看出肯定是儿子内部解决了比较好,到时最后
g
1
g_1
g1 一坨全部消不了就不优了。
那我们就先考虑你枚举儿子 y y y,它会有一些向上匹配的 g y g_y gy,那当然如果 g y > a x g_y>a_x gy>ax( a x a_x ax 是 x x x 的点权),那也只能浪费掉,至多拿 a x a_x ax 个。
接着我们就得出了从下面来的数量
n
u
m
num
num,那接着进行一个小小的分类讨论。
如果
n
u
m
>
2
a
x
num> 2a_x
num>2ax,那可以能匹配的都匹配了,不能匹配的也只能扔掉,所以匹配的是
2
a
x
2a_x
2ax,
g
x
=
0
g_x=0
gx=0。
如果
a
x
⩽
n
u
m
⩽
2
a
x
a_x\leqslant num\leqslant2a_x
ax⩽num⩽2ax,那我们可考虑它们能匹配就匹配,不能就拎上去匹配,然后一共会有
2
a
x
2a_x
2ax 条分支,所以当前的
n
u
m
num
num 个都省了,然后
g
x
=
2
a
x
−
n
u
m
g_x=2a_x-num
gx=2ax−num。
如果
n
u
m
<
a
x
num<a_x
num<ax,那它还是跟别人匹配吧,自己匹配太不优的,不过毕竟跟别人匹配了,还是少了
n
u
m
num
num 的删除,然后
g
x
=
n
u
m
g_x=num
gx=num。
然后 DP 出来用原来点权和减去 f 1 f_1 f1 即可。
代码
#include<cstdio>
#include<iostream>
#define ll long long
using namespace std;
struct node {
int to, nxt;
}e[400001];
int n, KK;
int x, y, le[200001];
ll f[200001], g[200001], ans, a[200001];
void add(int x, int y) {
e[++KK] = (node){y, le[x]}; le[x] = KK;
e[++KK] = (node){x, le[y]}; le[y] = KK;
}
void dfs(int now, int father) {//树形DP
ll num = 0;
for (int i = le[now]; i; i = e[i].nxt)
if (e[i].to != father) {
dfs(e[i].to, now);
f[now] += f[e[i].to];
num += min(1ll * a[now], g[e[i].to]);
}
f[now] += min(num, a[now] * 2);
if (num > a[now] * 2) g[now] = 0;
else if (num < a[now]) g[now] = a[now];
else g[now] = a[now] * 2 - num;
}
int main() {
// freopen("snow.in", "r", stdin);
// freopen("snow.out", "w", stdout);
scanf("%d", &n);
for (int i = 1; i <= n; i++) scanf("%lld", &a[i]), ans += a[i];
for (int i = 1; i < n; i++) {
scanf("%d %d", &x, &y);
add(x, y);
}
dfs(1, 0);
printf("%lld", ans - f[1]);
return 0;
}