【ybt高效进阶5-4-2】结点覆盖（树形DP）

结点覆盖

题目链接：ybt高效进阶5-4-2

题目大意

给你一棵树，选一个点有各自的费用。
然后选一个点可以覆盖它，它的儿子和它的父亲。
问你要覆盖整一个树的最小费用。

思路

一个点能覆盖自己父亲儿子，那一个点一定会被它自己或者父亲或者儿子覆盖到。
设 $f_{i,0/1/2}$ 为搞定 $i$ 的子树，$i$ 是被父亲 / 自己 / 儿子覆盖到的。

然后如果是自己覆盖记得要加上自己的费用，后面就随意。
然后父亲就是儿子除了选父亲都可以随便。

然后儿子的话就是儿子也是不能选父亲，而且要至少有一个选自己。
那你就看选儿子变成选自己损失最小的那个，让那个选。

然后这样转移就好啦，最后输出的就是根节点被自己或儿子占领的最小费用。（根节点没有父亲）

代码

#include<cstdio>
#include<iostream>
#define INF 0x3f3f3f3f3f3f3f3f

using namespace std;

struct node {
	int to, nxt;
}e[1501];
int n, a[1501], le[1501];
int x, y, z, fa[1501], KK;
int rt, f[1501][3];

void add(int x, int y) {
	e[++KK] = (node){y, le[x]}; le[x] = KK;
}

void dfs(int now, int father) {
	f[now][1] = a[now];
	int minn = INF;
	for (int i = le[now]; i; i = e[i].nxt)
		if (e[i].to != father) {
			dfs(e[i].to, now);
			f[now][0] += min(f[e[i].to][1], f[e[i].to][2]);
			f[now][1] += min(f[e[i].to][0], min(f[e[i].to][1], f[e[i].to][2]));
			f[now][2] += min(f[e[i].to][1], f[e[i].to][2]);
			minn = min(minn, f[e[i].to][1] - min(f[e[i].to][1], f[e[i].to][2]));
			//觉得选哪个儿子损失最小
		}
	f[now][2] += minn;//一定要选一个儿子
}

int main() {
	scanf("%d", &n);
	for (int i = 1; i <= n; i++) {
		scanf("%d", &x);
		scanf("%d", &a[x]);
		scanf("%d", &y);
		while (y--) {
			scanf("%d", &z);
			fa[z] = x; add(x, z);
		}
	}
	
	for (int i = 1; i <= n; i++)
		if (!fa[i]) {
			rt = i; break;
		}
	dfs(rt, 0);
	
	printf("%d", min(f[rt][1], f[rt][2]));
	
	return 0;
}