【树形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个点染 ...
随机推荐
- gogs私有代码库上传项目
https://blog.csdn.net/zhouxueli32/article/details/80538017 一.上传 在cmd命令里进入该项目 然后依次输入以下命令 git initgit ...
- 部门工资前三高的所有员工 - LeetCode
Employee 表包含所有员工信息,每个员工有其对应的工号 Id,姓名 Name,工资 Salary 和部门编号 DepartmentId . +----+-------+--------+---- ...
- Python课程第六天作业
1.以自己的理解总结为什么会出现循环导入,并用代码举例说明 循环导入报错并不是应为相互之间引用而导致报错,实际上是在导入一个模块时发现引用的模块不存在 示例如下: m1.py from m2 impo ...
- Redis-1-简介与安装
目录 1.Redis 简介 2.安装Redis 1.安装gcc redis是c语言编写的 2.下载redis安装包,在root目录下执行 3.解压redis安装包 4.进入redis目录 5.编译安装 ...
- 【linux】查看某个进程PID对应的文件句柄数量,查看某个进程当前使用的文件句柄数量
================================ 1.linux所有句柄查询 lsof -n|awk '{print $2}'|sort|uniq -c |sort -nr|more ...
- C# 通过反射获取winform上的控件
比如获取Button按钮: System.Reflection.FieldInfo[] fieldInfo = form.GetType().GetFields(System.Reflection.B ...
- C#反射_两合并更新实体
#region 更新实体模型 /// <summary> /// 更新实体模型 /// </summary> /// <typeparam name="T&qu ...
- python 绘图与可视化 Graphviz 二叉树 、 error: Microsoft Visual C++ 14.0 is required
需要对二叉树的构建过程进行可视化,发现了这个Graphviz软件,他对描绘数据间的关系十分擅长. 下载链接:https://graphviz.gitlab.io/_pages/Download/Dow ...
- 通俗易懂的讲解一下Java的代理模式
一.基本概念 代理模式是对象的结构模式. 代理模式给某一个对象提供一个代理对象,并由代理对象控制对原对象的引用(接口的引用) 二.静态代理 静态代理是指,代理类在程序运行前就已经定义好,其与**目标类 ...
- 深入理解Session和Cookie的区别
Cookie简介 Cookie意为“甜饼”,是由W3C组织提出,最早由Netscape社区发展的一种机制. 目前Cookie已经成为标准,所有的主流浏览器如IE.Netscape.Firefox.Op ...