树的最小子节点-算法题

tarespan

于 2021-12-16 01:17:12 发布

阅读量368

点赞数

文章标签：算法深度优先图论

本文链接：https://blog.csdn.net/weixin_43976081/article/details/121962804

版权

算法题专栏收录该内容

42 篇文章 0 订阅

订阅专栏

该博客介绍了一种使用深度优先搜索（DFS）并结合记忆化技术来解决在给定节点关系和价值的情况下，找到树中每个节点的最小子节点的方法。通过递归地遍历树的节点，并比较子节点的值，确定每个节点的最小子节点。输入包括节点关系和对应节点的值，输出是每个节点的最小子节点的值。

摘要由CSDN通过智能技术生成

题意：给定n个节点，标号为0~n-1，每个节点都有一个value值，求这棵树上每个节点的最小子节点；

输入：relation = [[1,0],[2,1],[3,1],[3,7],[4,3],[5,3],[6,3]], value = [3,2,5,4,6,1,7,0]
输出：[5,5,2,5,4,5,6,7]

输入：relation = [], value = [0]
输出：[0]

题解一：DFS+记忆化

vector<int> minSonNode(vector<vector<int>> &relation, vector<int> &value) {
    int n = value.size();
    vector<vector<int>> g(n);
    for (vector<int> &a : relation)
        g[a[1]].emplace_back(a[0]);
    vector<int> ret(n, -1);
    function<void(int)> dfs = [&](int x) {
        if (ret[x] != -1)
            return;
        ret[x] = x;
        for (int &y : g[x]) {
            dfs(y);
            ret[x] = (value[ret[y]] < value[ret[x]]) ? ret[y] : ret[x];
        }
    };
    for (int i = 0; i < n; ++i)
        dfs(i);
    return ret;
}