当前位置: 代码迷 >> 综合 >> caioj 1111 树形动态规划(TreeDP)6: 皇宫看守 (状态设计)
  详细解决方案

caioj 1111 树形动态规划(TreeDP)6: 皇宫看守 (状态设计)

热度:141   发布时间:2023-09-20 19:21:41.0

这道题的难点在于状态怎么设计
这道题要求全部都是安全的,所以我们做的时候自底向上每一个结点都要是安全的
结合前一题当前结点选和不选,我们可以分出四种情况出来
选 安全
选 不安全
不选 安全
不选 不安全

显然选 不安全是不可能的,那么就去掉
所以我们就可以设计状态为
caioj 1111 树形动态规划(TreeDP)6: 皇宫看守 (状态设计)表示i放人且安全
caioj 1111 树形动态规划(TreeDP)6: 皇宫看守 (状态设计)表示i不放人且安全
caioj 1111 树形动态规划(TreeDP)6: 皇宫看守 (状态设计)表示i不放人且不安全

那么状态转移方程最关键的就是怎么保证回溯的时候都是安全的。
我们只考虑以u为结点的子树,不考虑i的父亲
我们要让u的子树除了u以外全部是安全的,u自己安全和不安全分开讨论

对于 caioj 1111 树形动态规划(TreeDP)6: 皇宫看守 (状态设计), u放人且安全
那么显然这时儿子无论如何都是安全的(就算原来他是不安全的)
那么有caioj 1111 树形动态规划(TreeDP)6: 皇宫看守 (状态设计)
这里v是u的儿子,这时三种情况都可以,取最小

对于caioj 1111 树形动态规划(TreeDP)6: 皇宫看守 (状态设计) i不放人且不安全
那么为了保证u不安全肯定儿子不能放人,而这时我们要保证
儿子都安全,所以caioj 1111 树形动态规划(TreeDP)6: 皇宫看守 (状态设计)

对于caioj 1111 树形动态规划(TreeDP)6: 皇宫看守 (状态设计) i不放人且安全
儿子一定要安全的话有
caioj 1111 树形动态规划(TreeDP)6: 皇宫看守 (状态设计)
但是要保证u安全,v中至少有一个放人
这就比较麻烦了,我们要专门来判断v中有没有放人
如果没有的话,就加上caioj 1111 树形动态规划(TreeDP)6: 皇宫看守 (状态设计)
也就是以最小的费用使一个儿子从不放人到放人

最后还有一个小细节,之前我写树形dp搜到叶子都直接return的
这里不行,因为这时caioj 1111 树形动态规划(TreeDP)6: 皇宫看守 (状态设计)是不存在的(不考虑u的父亲)
所以这时要把caioj 1111 树形动态规划(TreeDP)6: 皇宫看守 (状态设计)初始化为最大值(代码中体现为最后加上caioj 1111 树形动态规划(TreeDP)6: 皇宫看守 (状态设计)

#include<cstdio>
#include<vector>
#include<algorithm>
#define REP(i, a, b) for(int i = (a); i < (b); i++)
using namespace std;const int MAXN = 2123;
int f[3][MAXN], a[MAXN], b[MAXN], n;
vector<int> g[MAXN];void dfs(int u)
{f[0][u] = a[u]; f[1][u] = f[2][u] = 0; 	int mint = 1e8, ok = 0;REP(i, 0, g[u].size()){int v = g[u][i];dfs(v);f[0][u] += min(f[0][v], min(f[1][v], f[2][v]));f[1][u] += min(f[0][v], f[1][v]);if(f[0][v] <= f[1][v]) ok = 1;mint = min(mint, f[0][v] - f[1][v]);f[2][u] += f[1][v];}if(!ok) f[1][u] += mint;
} int main()
{scanf("%d", &n);REP(i, 1, n + 1) {int u, k, son; scanf("%d", &u);scanf("%d%d", &a[u], &k);REP(j, 0, k){scanf("%d", &son);g[u].push_back(son);b[son] = 1;}	}REP(i, 1, n + 1)if(!b[i]){dfs(i);printf("%d\n", min(f[0][i], f[1][i]));break;}return 0;	
} 

 

  相关解决方案