【树形DP】【P3177】[HAOI2015] 树上染色
Description
给定一棵 \(n\) 个点的带权树,要求选 \(k\) 个点染成黑色,剩下染成白色,最大化两两同色点之间的距离和。
Limitations
\(0 \leq k \leq n \leq 2000\)
Solution
首先看一个trick:
考虑如下遍历一棵树的伪代码:
func dfs(u):
size[u] <- 1
for v in clild[u] do
dfs(v)
for i = 1 : size[u] do
for j = 1: size[v] do
do sth
end
end
size[u] <- size[u] + size[v]
end
end func
如果认为do sth的复杂度是 \(O(T)\) 的话,那么上述伪代码的时间复杂度是 \(O(n^2T)\),也就是说上述遍历整棵树外加 \(3\) 层 for 循环的时间复杂度是 \(O(n^2)\)。
证明上可以考虑如果记录每个点的 dfs 序,\(i\) 可以看作枚举 \(u\) 所有已经到达过的除 \(v\) 及其子树以外的后代。\(j\) 可以看作枚举 \(v\) 的后代。那么考虑任何一对点都只会在他们的 LCA \(u\) 处被枚举到,即每对点只会枚举一次,且 \(i\) 的 dfs 序一定小于 \(j\) 的 dfs 序,由于整个 dfs 序序列的顺序对个数又 \(O(n^2)\) 个,所以内层两层循环的总次数是 \(O(n^2)\) 的。于是上述代码的时间复杂度是 \(O(n^2T)\),其中 do sth 的时间复杂度是 \(O(T)\)
回到这个题,最直接的DP是设 \(f_{u, i}\) 为以 \(u\) 为根的子树,选了 \(i\) 个黑点的最优答案,但是发现无法转移,因为全局最优解和局部最优没什么关系,于是考虑像 [NOI2019]回家的路 一样,对贡献去做DP。
考虑将上述状态更换为以 \(u\) 为根的子树,选了 \(i\) 个黑点对答案的最大贡献是多少,在转移时考虑每个子树的贡献即可。
具体的,设 \(f_{u, i, j}\) 为以 \(u\) 为根的子树,考虑前 \(i\) 个孩子,选了 \(j\) 个黑点的最大贡献。
那么转移显然:
\]
其中 \(val\) 是这条边对答案造成的贡献,具体为边两侧黑点个数的乘积加上两侧白点个数的乘积的和再乘上边权。
例如子树中选择了 \(x\) 个黑点,那么贡献为
\]
其中 \(e_w\) 为边权。
考虑上述转移中,\(u\) 的状态 \(i\) 只依赖于状态 \(i-1\) 和 \(size_v\),因此可以滚动数组。滚动后的状态转移方程为:
\]
再转移时需要保证 \(f_{u,h}\) 是没有被 \(v\) 更新过的答案,也就是说比 \(j\) 小的状态应该在 \(j\) 之后更新,因此倒序枚举 \(j\) 即可。
在考虑最后一个问题:我们设计的方程是填表转移,即枚举了上面伪代码中 \(i\) 和 \(j\) 的和以及 \(i\),只有这样才能保证方程中 \(j\) 的转移是单调的从而滚动数组,因此我们需要把上述伪代码改成外层枚举两数和,内层枚举除 \(v\) 以外的已遍历过的子树的形式。
func dfs(u):
size[u] <- 1
for v in clild[u] do
dfs(v)
size[u] <- size[u] + size[v]
for sum = 1 : size[u] do
for i = max(0, sum - size[v]) : size[v] do
do sth
end
end
end
end func
虽然写成这样的时间复杂度十分难以分析,但是枚举和再枚举其中一个的复杂度显然和枚举两个求和的复杂度相同,因此上述代码的时间复杂度也为 \(O(n^2T)\)
由于单次转移是 \(O(1)\) 的,因此算法的总时间复杂度 \(O(n^2)\)
Code
#include <cstdio>
#include <cstring>
#include <algorithm>
const int maxn = 2003;
int n, K, dK;
int sz[maxn];
ll frog[maxn][maxn];
struct Edge {
int v, w;
Edge *nxt;
Edge(const int _v, const int _w, Edge *h) : v(_v), w(_w), nxt(h) {}
};
Edge *hd[maxn];
void dfs(const int u, const int fa);
int main() {
freopen("1.in", "r", stdin);
qr(n); qr(K); dK = n - K;
for (int i = 1, u, v, w; i < n; ++i) {
u = v = w = 0; qr(u); qr(v); qr(w);
hd[u] = new Edge(v, w, hd[u]);
hd[v] = new Edge(u, w, hd[v]);
}
dfs(1, 0);
qw(frog[1][K], '\n', true);
return 0;
}
void dfs(const int u, const int fa) {
sz[u] = 1;
memset(frog[u] + 2, -1, 16008);
for (auto e = hd[u]; e; e = e->nxt) if (e->v != fa) {
int v = e->v; dfs(v, u); sz[u] += sz[v];
for (int i = std::min(sz[u], K); ~i; --i) {
for (int j = i, lim = std::max(0, i - sz[v]); j >= lim; --j) if (~frog[u][j]) {
int k = i - j; if (frog[v][k] == -1) continue;
ll val = (1ll * k * (K - k) + 1ll * (sz[v] - k) * (dK - sz[v] + k)) * e->w;
frog[u][i] = std::max(frog[u][i], frog[u][j] + frog[v][k] + val);
}
}
}
}
Summary
1、上面那种神奇的枚举方式遍历整棵树的时间复杂度是 \(O(n^2)\) 的
2、当问题不满足最优子结构时,可以考虑DP每个子问题对答案的贡献。
3、很多时候刷表法难以滚动数组,需要转化成填表法
appreciation
感谢 @DDOSvoid 大爷与我的讨论以及对我的启发
【树形DP】【P3177】[HAOI2015] 树上染色的更多相关文章
- 洛谷 P3177 [HAOI2015]树上染色 树形DP
洛谷 P3177 [HAOI2015]树上染色 树形DP 题目描述 有一棵点数为 \(n\) 的树,树边有边权.给你一个在 \(0 \sim n\)之内的正整数 \(k\) ,你要在这棵树中选择 \( ...
- 洛谷P3177 [HAOI2015]树上染色(树形dp)
题目描述 有一棵点数为 N 的树,树边有边权.给你一个在 0~ N 之内的正整数 K ,你要在这棵树中选择 K个点,将其染成黑色,并将其他 的N-K个点染成白色 . 将所有点染色后,你会获得黑点两两之 ...
- 洛谷 P3177 [HAOI2015]树上染色
题目链接 题目描述 有一棵点数为 \(N\) 的树,树边有边权.给你一个在 \(0~ N\) 之内的正整数 \(K\) ,你要在这棵树中选择 \(K\)个点,将其染成黑色,并将其他 的\(N-K\)个 ...
- Luogu P3177 [HAOI2015]树上染色
一道有机结合了计数和贪心这一DP两大考点的神仙题,不得不说做法是很玄妙. 首先我们很容易想到DP,设\(f_{i,j}\)表示在以\(i\)为根节点的子树中选\(j\)个黑色节点的最大收益值. 然后我 ...
- 【洛谷】P3177 [HAOI2015]树上染色
懒得复制题面了直接传送门吧 分析 直接求点与点之间的距离感觉不是很好求,所以我们考虑换一个求法. 瞄了一眼题解 距离跟路径上边的长度有关,所以我们直接来看每一条边的贡献吧(这谁想得到啊) 对于每一条边 ...
- P3177 [HAOI2015]树上染色
题目描述 有一棵点数为 N 的树,树边有边权.给你一个在 0~ N 之内的正整数 K ,你要在这棵树中选择 K个点,将其染成黑色,并将其他 的N-K个点染成白色 . 将所有点染色后,你会获得黑点两两之 ...
- 洛谷P3177 [HAOI2015]树上染色(树上背包)
题意 题目链接 Sol 比较套路吧,设\(f[i][j]\)表示以\(i\)为根的子树中选了\(j\)个黑点对答案的贡献 然后考虑每条边的贡献,边的两边的答案都是可以算出来的 转移的时候背包一下. # ...
- BZOJ 4033: [HAOI2015]树上染色题解
BZOJ 4033: [HAOI2015]树上染色题解(树形dp) 标签:题解 阅读体验:https://zybuluo.com/Junlier/note/1327400 原题地址: BZOJ 403 ...
- bzoj 4033: [HAOI2015]树上染色 [树形DP]
4033: [HAOI2015]树上染色 我写的可是\(O(n^2)\)的树形背包! 注意j倒着枚举,而k要正着枚举,因为k可能从0开始,会使用自己更新一次 #include <iostream ...
- 【BZOJ4033】[HAOI2015]树上染色 树形DP
[BZOJ4033][HAOI2015]树上染色 Description 有一棵点数为N的树,树边有边权.给你一个在0~N之内的正整数K,你要在这棵树中选择K个点,将其染成黑色,并将其他的N-K个点染 ...
随机推荐
- CentOS中设置Apache服务器网站访问日志[每天的日志]
在阿里云的linux 服务器下Apache的日志默认设置是七天更新一次, 并且所在的目录无法通过FTP浏览器查看, 这样让小白操作起来非常麻烦 可以使用rotatelogs来设置服务器的网站访问日志按 ...
- Mysql 错误:Duplicate entry '0' for key 'PRIMARY'
[1]添加数据报错:Duplicate entry '0' for key 'PRIMARY' (1)问题现象 SQL 语句如下: DROP TABLE test_distinct; CREATE T ...
- KVM 学习笔记
查看虚拟化环境 (1)查看虚拟机环境 (2)查看kvm模块支持 (3)查看虚拟工具版本 (4)查看网桥
- KVM虚拟机网络配置 Bridge方式,NAT方式
https://blog.csdn.net/hzhsan/article/details/44098537/
- pyhanlp的安装
github 的官方地址:https://github.com/hankcs/pyhanlp conda install -c conda-forge jpype1 pip install pyhan ...
- MySQL语言分类——DDL
DDL的全称Data Definition Language,即数据定义语言 DDL的语法有:create.alter.drop.rename.truncate.对此做一个详细的解释: create ...
- vue 的 Class 与 Style 绑定
操作元素的 class 列表和内联样式是数据绑定的一个常见需求.因为它们都是属性,所以我们可以用 v-bind 处理它们:只需要通过表达式计算出字符串结果即可.不过,字符串拼接麻烦且易错.因此,在将 ...
- Win10 默认用Windows照片查看程序打开图片
::复制以下内容到记事本: @echo off&cd\&color 0a&cls echo 恢复Win10照片查看器 reg add "HKLM\SOFTWARE\M ...
- Linux安装node环境
一.进行连接远程: 1.命令窗口 —> 输入 ssh 用户名@主机IP —> 回车 2.输入密码 (输入后回车) 3.进入根目录 (命令:cd / ) 二.Linux环境安装node: T ...
- 大规模定制模式之于MES的三点思考
大规模定制(Mass Custermization) ,其目标是大规模生产定制化产品,并且在效率.质量(一致性)等指标方面与大规模批量生产等齐. 这是一种理想或者追求,其提出的背景是目前越发普遍的多品 ...