树+贪心

黑白树

https://ac.nowcoder.com/acm/problem/13249?&headNav=acm

  • 题目描述

    一棵n个点的有根树,1号点为根,相邻的两个节点之间的距离为1。树上每个节点i对应一个值k[i]。每个点都有一个颜色,初始的时候所有点都是白色的。
    你需要通过一系列操作使得最终每个点变成黑色。每次操作需要选择一个节点i,i必须是白色的,然后i到根的链上(包括节点i与根)所有与节点i距离小于k[i]的点都会变黑,已经是黑的点保持为黑。问最少使用几次操作能把整棵树变黑。

  • 输入描述:

    第一行一个整数n (1 ≤ n ≤ 10^5)
    接下来n-1行,每行一个整数,依次为2号点到n号点父亲的编号。
    最后一行n个整数为k[i] (1 ≤ k[i] ≤ 10^5)

  • 样例解释:

    对节点3操作,导致节点2与节点3变黑
    对节点4操作,导致节点4变黑
    对节点1操作,导致节点1变黑
    无序列表内容

  • 输出描述:

    一个数表示最少操作次数

解题思路

题目一上手,可以简单画个图,通过树图,发现最后一个点是必选的,那叶子节点上方的点我们应该选择那些点呢?是不是选择叶子节点长度够不到的点就选呢?不是的,我们可以假设叶子节点k=3,它父节点k=10,其余节点k值都是1,上方贪心方法直接裂开。不过我们还是可以通过这个错误的贪心方法发现问题,加以修改就是我们正确的贪心攻略。对于每一个节点我们要维护题目中的k数组表示子节点到目前节点往上可以走最远的距离,也要新开辟一个f数组,维护来自子节点往上最远的覆盖距离,当这个覆盖距离为0时,说明这个点无法覆盖,我们需要新增一个节点,我们选取的点应该是越过这个点可以往上走最远的节点涂黑。

Code

#include <bits/stdc++.h>
using namespace std;
#define js ios::sync_with_stdio(false);cin.tie(0); cout.tie(0)
typedef long long ll;

inline int read() {
    int s = 0, w = 1; char ch = getchar();
    while (ch < 48 || ch > 57) { if (ch == '-') w = -1; ch = getchar(); }
    while (ch >= 48 && ch <= 57) s = (s << 1) + (s << 3) + (ch ^ 48), ch = getchar();
    return s * w;
}

const int N = 1e5 + 7;
vector<int> p[N];
int k[N], f[N]; //需要维护的2个数组
int ans;

void dfs(int x, int fa) {
    for (int i = 0; i < p[x].size(); ++i) {
        int u = p[x][i];
        dfs(u, x); //注意k数组,我们不关心这个最远来自谁,只要知道可以最远多少就行了
        k[x] = max(k[x], k[u] - 1); //维护k[x] 是保持还是子节点消耗1,更新取max
        f[x] = max(f[x], f[u] - 1); // 还可以往上延申长度
    }
    if (!f[x]) { //如果子节点都无法覆盖到这个点
        ++ans; //选取这个点
        f[x] = k[x]; //更新这个点的f[x]
    }
}

int main() {
    int n = read();
    for (int i = 2; i <= n; ++i) {
        int u = read();
        p[u].push_back(i); // = = 注意这个父子关系别搞反了
    }
    for (int i = 1; i <= n; ++i)    k[i] = read();
    dfs(1, 0);
    printf("%d\n", ans);
    return 0;
}
每日一题 文章被收录于专栏

每日一题

全部评论
你在题解举的那个例子,有问题把,叶子节点k=3,那么它的父节点,一定会被它影响而变黑,因为,题目说的选的点必须是白色的,说明我们不能选叶子节点的父节点,尽管它的父节点k=10,也不能选择,我是一格菜鸡,无法达到大神的级别。如果有错误,勿喷
点赞 回复 分享
发布于 2022-03-03 20:49
膜拜
点赞 回复 分享
发布于 2020-04-08 16:25

相关推荐

自从我室友在计算机导论课上听说了“刷&nbsp;LeetCode&nbsp;是进入大厂的敲门砖”,整个人就跟走火入魔了一样。他在宿舍门口贴了一张A4纸,上面写着:“正在&nbsp;DP,请勿打扰,否则&nbsp;Time&nbsp;Limit&nbsp;Exceeded。”日记本的扉页被他用黑色水笔加粗描了三遍:“Talk&nbsp;is&nbsp;cheap.&nbsp;Show&nbsp;me&nbsp;the&nbsp;code。”连宿舍聚餐,他都要给我们讲解:“今天的座位安排可以用回溯算法解决,但为了避免栈溢出,我建议用动态规划。来,这是状态转移方程:dp[i][j]&nbsp;代表第&nbsp;i&nbsp;个人坐在第&nbsp;j&nbsp;个位置的最优解。”我让他去楼下取个快递,他不直接去,非要在门口踱步,嘴里念念有词:“这是一个图的遍历问题。从宿舍楼(root)到驿站(target&nbsp;node),我应该用&nbsp;BFS&nbsp;还是&nbsp;DFS?嗯,求最短路径,还是广度优先好。”和同学约好出去开黑,他会提前发消息:“集合点&nbsp;(x,&nbsp;y),我们俩的路径有&nbsp;k&nbsp;个交点,为了最小化时间复杂度,应该在&nbsp;(x/2,&nbsp;y/2)&nbsp;处汇合。”有一次另一个室友低血糖犯了,让他帮忙找颗糖,他居然冷静地分析道:“别急,这是一个查找问题。零食箱是无序数组,暴力查找是&nbsp;O(n)。如果按甜度排序,我就可以用二分查找,时间复杂度降到&nbsp;O(log&nbsp;n)。”他做卫生也要讲究算法效率:“拖地是典型的岛屿问题,要先把连通的污渍区块都清理掉。倒垃圾可以用双指针法,一个指针从左往右,一个从右往左,能最快匹配垃圾分类。”现在我们宿舍的画风已经完全变了,大家不聊游戏和妹子,对话都是这样的:“你&nbsp;Two&nbsp;Sum&nbsp;刷了几遍了?”“别提了,昨天遇到一道&nbsp;Hard&nbsp;题,我连暴力解都想不出来,最后只能看题解。你呢?”“我动态规划还不行,总是找不到最优子结构。今天那道接雨水给我整麻了。”……LeetCode&nbsp;真的害了我室友!!!
老六f:编程嘉豪来了
AI时代还有必要刷lee...
点赞 评论 收藏
分享
评论
10
1
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务