8.6 KiB
AcWing
1077
. 皇宫看守
一、题目描述
太平王世子事件后,陆小凤成了皇上特聘的御前一品侍卫。
皇宫各个宫殿的分布,呈一棵树的形状,宫殿可视为树中节点,两个宫殿之间如果存在道路直接相连,则该道路视为树中的一条边。
已知,在一个宫殿镇守的守卫不仅能够观察到本宫殿的状况,还能 观察到与该宫殿直接存在道路相连的其他宫殿的状况。
大内保卫森严,三步一岗,五步一哨,每个宫殿都要有人全天候看守,在不同的宫殿安排看守所需的费用不同。
可是陆小凤手上的经费不足,无论如何也没法在每个宫殿都安置留守侍卫。
帮助陆小凤布置侍卫,在看守全部宫殿的前提下,使得花费的经费最少。
输入格式 输入中数据描述一棵树,描述如下:
第一行 n
,表示树中节点的数目。
第二行至第 n+1
行,每行描述每个宫殿节点信息,依次为:该宫殿节点标号 i
,在该宫殿安置侍卫所需的经费 k
,该节点的子节点数 m
,接下来 m
个数,分别是这个节点的 m
个子节点的标号 r_1,r_2,…,r_m
。
对于一个 n
个节点的树,节点标号在 1
到 n
之间,且标号不重复。
输出格式 输出一个整数,表示最少的经费。
数据范围
1≤n≤1500
输入样例:
6
1 30 3 2 3 4
2 16 2 5 6
3 5 0
4 4 0
5 11 0
6 5 0
输出样例:
25
样例解释:
在2、3、4
节点安排护卫,可以观察到全部宫殿,所需经费最少,为 16 + 5 + 4 = 25
。
二、对比
和 Acwing
323
. 战略游戏 这题非常类似,但又有些不同
Acwing
323
. 战略游戏在一条道路的两个节点至少有一个是放的,而这题不一定,比如

我们可以在3
,4
,5
,6
号点放,其他各点都不放。这样1-2
的两个端点都没有守卫,这就和 Acwing
323
. 战略游戏 这题不同了
三、思路
不过稍微观察可以发现,Acwing
323
. 战略游戏 这题的 切入点是边,而本题的 切入点是宫殿,也就是树上的节点。那么可以用 放不放 这个状态上 增加一个状态,即
-
当前节点不放守卫
- 父节点放守卫:
f[u][0]
- 子节点放守卫:
f[u][1]
- 父节点放守卫:
-
当前节点放置守卫:
f[u][2]
其中
u
为当前节点,这样就可以 涵盖整个状态空间
解释:状态描述这么划分,其实是 存在重叠情况的,比如
u
节点不放守卫,父节点放守卫:f[u][0]
,而且,它的子节点也放了守卫:f[u][1]
,这种情况是可能存在的:(f[u][0]
与f[u][1]
同时存在,不冲突!)。这样并不奇怪,因为我们最终要求的是最小值,只要不遗漏某种情况,最终的最小值是一样的
有了状态表示,接下来就是推导状态转移了,记j
为u
的子节点,那么
\displaystyle \large f[u][0]=\sum min\left \{f[v][1],f[v][2] \right \}
含义:当前节点u
不放且被父节点看到,那么子节点j
只能放或者被其子节点看到。 \displaystyle \large f[u][1]=f[k][2]+ \sum min\left \{ f[v][1],f[v][2] \right \} ~ k \neq j
含义:当前节点u
不放,u
的子节点k
放了,u
的其他子节点j
放(f[j][2]
)或者不放(即f[j][1]
),需要枚举k
。
没有
f[j][0]
是因为u
不放
\displaystyle \large f[u][2]=\sum min\left \{ f[v][0],f[v][1],f[v][2] \right \}
含义:当前节点放了,那么子节点取哪种状态都可以
1
,3
实现起来比较容易,至于2
的实现,有点麻烦,因为如果真的去一次次遍布计算是哪个k
儿子,效率就会很低。
不这样算,还能怎么算呢?
补集思想
- 把所有的
min(f[v][1],f[v][2])
都加在一起,不管v
是不是等于k
。 - 讨论自己不放守卫,枚举每个儿子视为好儿子,选中则
+f[v][2]
,同时, 其它儿子不靠父亲靠自己的代价总和就是sum(min(f[v][1],f[v][2]) - min(f[k][1],f[k][2])
,这样,算法就是O(1)
的速度
四、树形DP
代码
#include <bits/stdc++.h>
using namespace std;
const int INF = 0x3f3f3f3f;
const int N = 1510;
int n;
int h[N], e[N], ne[N], idx;
int w[N]; // 点的权值,不是边,不是边,不是边!
/*
状态表示:
(1)集合
f[u][0]:u号节点不放守卫,被父节点看守的方案
f[u][1]:u号节点不放守卫,被某个子节点看守的方案
f[u][2]:u号节点自己安排守卫看住的方案
(2)属性
花费最小值
*/
int f[N][3];
int in[N]; // 入度
// 邻接表
void add(int a, int b) {
e[idx] = b, ne[idx] = h[a], h[a] = idx++;
}
void dfs(int u) {
// 初始化
f[u][0] = 0; // 被父亲守护,对于u子树而言,代价为0
f[u][1] = INF; // 被某个子节点守护,但还没有评选中由哪个来看守更合适,预求最小先设最大
f[u][2] = w[u]; // 被自己守护,对于u子树而言,代价为w[u]
for (int i = h[u]; ~i; i = ne[i]) {
int j = e[i];
dfs(j);
// ① u不放守卫,被父节点守护,它的儿子们可以放守卫,也可以不放守卫,两者代价取min
f[u][0] += min(f[j][1], f[j][2]);
// ③ u放守卫,儿子可以指望父亲,可以自己放守卫,也可以指望某个孙子来守卫,三者代价取min
f[u][2] += min({f[j][0], f[j][1], f[j][2]});
}
// ② u不放守卫,由某个好儿子守护
for (int i = h[u]; ~i; i = ne[i]) {
int j = e[i];
// Q:f[u][0]是什么意思?
// 向上看第39行,本质上就是所有子节点j的最小花费值累加和,没有扣除掉孝顺儿子的花费值
// 假设j是孝顺儿子,需要在其中扣除掉j的min(f[j][1],f[j][2])贡献,同时,
// 它的代价跑不了,此时就不能取min了,而是实打实的f[j][2]
f[u][1] = min(f[u][1], f[j][2] + f[u][0] - min(f[j][1], f[j][2]));
}
}
int main() {
memset(h, -1, sizeof h); // 初始化
cin >> n; // 节点数
for (int i = 1; i <= n; i++) { // n个节点
int x, m; // 节点号,代价,子节点数
cin >> x >> w[x] >> m;
while (m--) {
int y;
cin >> y;
add(x, y); // 有向图
in[y]++; // 入度,用于找根
}
}
// 找根
int root = 1;
while (in[root]) root++;
// dfs
dfs(root);
// f[root][0]表示root不放守卫,被父节点看守,根节点是没有父节点的, 所以此状态是无效状态,不参与结果讨论
// 最终最小代价值取min(f[root][1],f[root][2])
printf("%d\n", min(f[root][1], f[root][2]));
return 0;
}
五、问题与解答
Q
:f[u][1]
为什么要单独再开一个for
来求,为什么不能直接在第一个for
里面同时求f[u][0]、f[u][2]
和f[u][1]
?
答:
注意第一个循环中的写法:
f[u][0] += min(f[v][1], f[v][2]);
这是在枚举每个u
的儿子v
,然后把v
儿子自己放守卫,v
儿子指望它的某个儿子看守望的情况两者的最小值进行累加,最终形成累加和sum=f[u][0]
,注意,是最终的 累加和 啊!!!也就是所有儿子v_1,v_2,v_3
,…的取值累加和。
在第二个循环中,我们期望如果现在指望 v
儿子给u
守卫的话,那么f[v][2]
是肯定要付出的代价,那其它儿子需要付出的整体代价就是
sum-min(f[v][1],f[v][2])
注意:这里的是sum
!!!
如果我们把三个状态转移全部放在第一个循环中,我们就会发现,当我们需要全部的累加和sum
时,上面的f[u][0]
还没有累加完毕!!还只是一部分儿子v
的sum
和,并不是全部,而我们要全部的sum
和,这就意味着我们需要等待上面的所有v
累加完,也就是执行完一遍循环后,才能拿到这个需要的sum
值,这就解释了为什么要用两遍循环的原因。