今天,接着BFS,按tag来。
Tags breadth-first-search | graph
最小高度树。
累了,上题目
题目
对于一个具有树特征的无向图,我们可选择任何一个节点作为根。图因此可以成为树,在所有可能的树中,具有最小高度的树被称为最小高度树。给出这样的一个图,写出一个函数找到所有的最小高度树并返回他们的根节点。
格式
该图包含 n 个节点,标记为 0 到 n - 1。给定数字 n 和一个无向边 edges 列表(每一个边都是一对标签)。
你可以假设没有重复的边会出现在 edges 中。由于所有的边都是无向边, [0, 1]和 [1, 0] 是相同的,因此不会同时出现在 edges 里。
示例 1:
输入: n = 4, edges = [[1, 0], [1, 2], [1, 3]]
0
|
1
/ \
2 3
输出: [1]
示例 2:
输入: n = 6, edges = [[0, 3], [1, 3], [2, 3], [4, 3], [5, 4]]
0 1 2
\ | /
3
|
4
|
5
输出: [3, 4]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/minimum-height-trees
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
累了,上代码
代码
class Solution:
def findMinHeightTrees(self, n: int, edges: List[List[int]]) -> List[int]:
if n==1: return [0]
degree = [0 for i in range(n)]
adj = [[] for i in range(n)]
for pre, cur in edges:
adj[pre].append(cur)
adj[cur].append(pre)
degree[pre] += 1
degree[cur] += 1
queue = []
for i, d in enumerate(degree):
if d == 1:
queue.append(i)
while queue:
temp_queue = []
for node in queue:
for n in adj[node]:
degree[n] -= 1
if degree[n] == 1:
temp_queue.append(n)
if not temp_queue:
return queue
queue = temp_queue
好了,就这样吧。多看看。
对了,注意n==1的情况就好了。