https://ac.nowcoder.com/acm/problem/13249
题意:
给一棵树,每个点有一个值 k[i],表示取这个点的话可以将这个点到根节点的所有点里,例句这个点距离小于 k[i] 的点变成黑色,求所有点都变成黑色最少取几个点。
解法:
贪心,从根节点向上遍历,如果这个结点的值等于0,那么我们取这个点,反之,不取这个点
所以维护一个数组 a,表示还能取的点的个数,然后遍历的同时维护一下即可。
注意可能存在取已经被染成黑色的结点,比如
贪心的话,首先取 4 号点,然后可以覆盖 3 号点,然后取 2 号点和 1 号点,但实际上我们取 4 号点和 3 号点即可。
所以在遍历的同时,再维护一个子树的最大 (k 值 - 到子树根节点距离) 即可
#include <bits/stdc++.h> #define ll long long #define sc scanf #define pr printf using namespace std; const int MAXN = 1e5 + 5; struct edge { int to; int nex; }e[MAXN * 2]; int head[MAXN], tot; void init() { memset(head, -1, sizeof(head)); tot = 1; } void add(int a, int b) { e[tot] = edge{ b,head[a] }; head[a] = tot++; } int k[MAXN], a[MAXN]; int ans; void dfs(int u, int fa) { a[u] = 0; for (int i = head[u]; i + 1; i = e[i].nex) { int v = e[i].to; if (v == fa) continue; dfs(v, u); a[u] = max(a[u], a[v] - 1); k[u] = max(k[u], k[v] - 1); } if (a[u] == 0) { ans++; a[u] = k[u]; } } int main() { init(); int n; sc("%d", &n); for (int i = 2; i <= n; i++) { int t; sc("%d", &t); add(i, t); add(t, i); } for (int i = 1; i <= n; i++) sc("%d", &k[i]); dfs(1, 0); pr("%d", ans); }