开启我的LeetCode刷题日记:310. 最小高度树

79 阅读3分钟

一起养成写作习惯!这是我参与「掘金日新计划 · 4 月更文挑战」的第6天,点击查看活动详情

编程世界总是离不了算法

最近在看框架源码时,会有很多算法的实现逻辑,有时候会感到吃力

于是决定蹭着假期,加强算法和数据结构相关的知识

那怎么提升呢?

其实我知道算法这东西没有捷径,多写多练才能提升,于是我开启我的LeetCode刷题之旅

第一阶段目标是:200道,每天12

为了不乱,本系列文章目录分为三部分:

  1. 今日题目:xxx
  2. 我的思路
  3. 代码实现

今天题目:310. 最小高度树

树是一个无向图,其中任何两个顶点只通过一条路径连接。 换句话说,一个任何没有简单环路的连通图都是一棵树。

给你一棵包含 n 个节点的树,标记为 0 到 n - 1 。给定数字 n 和一个有 n - 1 条无向边的 edges 列表(每一个边都是一对标签),其中 edges[i] = [ai, bi] 表示树中节点 ai 和 bi 之间存在一条无向边。

可选择树中任何一个节点作为根。当选择节点 x 作为根节点时,设结果树的高度为 h 。在所有可能的树中,具有最小高度的树(即,min(h))被称为 最小高度树 。

请你找到所有的 最小高度树 并按 任意顺序 返回它们的根节点标签列表。

树的 高度 是指根节点和叶子节点之间最长向下路径上边的数量。  

示例 1:

输入:n = 4, edges = [[1,0],[1,2],[1,3]] 输出:[1] 解释:如图所示,当根是标签为 1 的节点时,树的高度是 1 ,这是唯一的最小高度树。

示例 2:

输入:n = 6, edges = [[3,0],[3,1],[3,2],[3,4],[5,4]] 输出:[3,4]

我的思路

解题思路

使用一个map存储整棵树,key为每个节点,value为节点所能连接到的节点的key组成的一个set。 使用这样的方式是因为map和set可以很方便地获取到里面元素个数的同时还可以直接进行删除操作而无需消耗额外时间。(使用object无法直接获取个数,使用array删除前面元素时会消耗额外时间) 然后用一个数组来存每次需要删的节点,然后遍历数组,对节点删除前,对于该节点连接到的其它节点进行“切断链接”的操作,若切断后该节点剩余连接的节点数变为1,将会被存到新的newArr中在下一次循环中再次被“剪枝”掉。 直到tree中只剩两个或者一个节点时,停止循环并输出剩余的节点即可。

代码实现

/**
 * @param {number} n
 * @param {number[][]} edges
 * @return {number[]}
 */
var findMinHeightTrees = function(n, edges) {
    if (n === 1) return [0];
    const tree = new Map();
    edges.forEach((line) => {
        const [p1, p2] = line;
        if (tree.has(p1)) {
            tree.get(p1).add(p2);
        } else {
            tree.set(p1, new Set([p2]));
        }
        if (tree.has(p2)) {
            tree.get(p2).add(p1);
        } else {
            tree.set(p2, new Set([p1]));
        }
    });

    // 维护一个叶节点队列
    let deleteArr = [];
    tree.forEach((value, key) => {
        if (value.size === 1) {
            deleteArr.push(key);
        }
    });

    while (deleteArr.length && tree.size > 2) {
        const newArr = [];
        deleteArr.forEach((item) => {
            tree.get(item).forEach((to) => {
                // 在找到所连接的节点切断其链接
                tree.get(to).delete(item);
                if (tree.get(to).size === 1) {
                    newArr.push(to);
                }
            });
            // 删除该节点
            tree.delete(item);
        });
        deleteArr = newArr;
    }
    const res = [];
    tree.forEach((value, key) => {
        res.push(key);
    });
    return res;
};



总结

实现方式其实有很多,这里仅供参考~

由于刚开始刷题,也不知道从哪里刷好,如果前辈们有好的建议,希望不吝赐教,感谢🌹