LeetCode 310. 最小高度树

一、题目

1、题目描述

  树是一个无向图，其中任何两个顶点只通过一条路径连接。 换句话说，一个任何没有简单环路的连通图都是一棵树。
  给你一棵包含 n 个节点的树，标记为 0 到 n - 1 。给定数字 n 和一个有 n - 1 条无向边的 edges 列表（每一个边都是一对标签），其中 edges[i] = [ai, bi] 表示树中节点 ai 和 bi 之间存在一条无向边。
可选择树中任何一个节点作为根。当选择节点 x 作为根节点时，设结果树的高度为 h 。在所有可能的树中，具有最小高度的树（即，min(h)）被称为 最小高度树 。
  请你找到所有的 最小高度树 并按 任意顺序 返回它们的根节点标签列表。
树的 高度 是指根节点和叶子节点之间最长向下路径上边的数量。
  样例输入： n = 4, edges = [[1,0],[1,2],[1,3]]
  样例输出： [1]

2、基础框架

• C++ 版本给出的基础框架代码如下：

class Solution {
public:
    vector<int> findMinHeightTrees(int n, vector<vector<int>>& edges) {

    }
};

3、原题链接

LeetCode 310. 最小高度树

二、解题报告

1、思路分析

  $(1)$ 根据给定的边，建立双向边；
  $(2)$ 求树的最长路：树形DP或者广搜，先从 0 搜到一个点 $a$ 代表离 0 最远的点，然后再从 $a$ 搜到的点 $b$，$b$ 为 离 $a$ 最远的点（这里的知识点是树的直径）。
  $(3)$ 对于 $a$ $b$ 这条路径记录所有路点，如果路点是奇数，返回路径中间的点，如果是偶数，返回中间的两个点。

2、时间复杂度

   最坏时间复杂度 $O(n)$ 。

3、代码详解

class Solution {
    #define maxn 100000
    vector <int> Edges[maxn];
    int pre[maxn];

    void init(int n) {
        for(int i = 0; i < n; ++i) {
            Edges[i].clear();
        }
    }

    void addEdge(int u, int v) {
        Edges[u].push_back(v);
        Edges[v].push_back(u);
    }

    int bfs(int n, int u) {
        int dist[maxn];
        int i;
        queue<int> q;
        memset(dist, -1, sizeof(dist));
        dist[u] = 0;
        pre[u] = -1;
        q.push(u);

        while(!q.empty()) {
            int u = q.front();
            q.pop();

            for(i = 0; i < Edges[u].size(); ++i) {
                int v = Edges[u][i];
                if(dist[v] == -1) {
                    dist[v] = dist[u] + 1;
                    pre[v] = u;
                    q.push(v);
                }
            }
        }
        int maxv = 0;
        for(i = 1; i < n; ++i) {
            if(dist[i] > dist[maxv]) {
                maxv = i;
            }
        }
        return maxv;
    }

public:
    vector<int> findMinHeightTrees(int n, vector<vector<int>>& edges) {
        int i;
        vector< int > path;
        init(n);
        for(i = 0; i < edges.size(); ++i) {
            addEdge(edges[i][0], edges[i][1]);
        }
        int x = bfs(n, 0);
        int y = bfs(n, x);

        // printf("%d %d\n", x, y);
        
        path.clear();
        while(y != x) {
            path.push_back(y);
            y = pre[y];
        }
        path.push_back(x);

        if(path.size() & 1) {
            return { path[ path.size()/2 ] };
        }else {
            return { path[ path.size()/2-1 ], path[ path.size()/2 ]  };

        }
    }
};

---

三、本题小知识

  树的直径就是树的最长路。

---

四、加群须知

  相信看我文章的大多数都是「 大学生 」，能上大学的都是「 精英 」，那么我们自然要「 精益求精 」，如果你还是「 大一 」，那么太好了，你拥有大把时间，当然你可以选择「 刷剧 」，然而，「 学好算法 」，三年后的你自然「 不能同日而语 」。
  那么这里，我整理了「 几十个基础算法 」 的分类，点击开启：

🌌《算法入门指引》🌌
  如果链接被屏蔽，或者有权限问题，可以私聊作者解决。

  大致题集一览：

  为了让这件事情变得有趣，以及「 照顾初学者 」，目前题目只开放最简单的算法 「 枚举系列 」 （包括：线性枚举、双指针、前缀和、二分枚举、三分枚举），当有 一半成员刷完 「 枚举系列 」 的所有题以后，会开放下个章节，等这套题全部刷完，你还在群里，那么你就会成为「 夜深人静写算法 」专家团 的一员。
  不要小看这个专家团，三年之后，你将会是别人 望尘莫及 的存在。如果要加入，可以联系我，考虑到大家都是学生， 没有「 主要经济来源 」，在你成为神的路上，「 不会索取任何 」。
  🔥联系作者，或者扫作者主页二维码加群，加入刷题行列吧🔥

---

🔥让天下没有难学的算法🔥
C语言免费动漫教程，和我一起打卡！ 🌞《光天化日学C语言》🌞
让你养成九天持续刷题的习惯 🔥《九日集训》🔥
入门级C语言真题汇总 🧡《C语言入门100例》🧡
组团学习，抱团生长 🌌《算法零基础100讲》🌌
几张动图学会一种数据结构 🌳《画解数据结构》🌳
竞赛选手金典图文教程 💜《夜深人静写算法》💜