博客
关于我
【Lintcode】1298. Minimum Height Trees
阅读量:227 次
发布时间:2019-02-28

本文共 2153 字,大约阅读时间需要 7 分钟。

题目地址:

给定一个 n n n阶无向图,其有树性质。若取其中某个顶点为根,其就成为一棵有根树,从而有树高。求使得树高最小的顶点编号。答案也许不唯一。

思路是BFS。像剥洋葱一样从最外层度为 1 1 1的顶点一层一层向里面剥,最后一层剥掉的皮就是答案。证明和详细解释可以参考。代码如下:

import java.util.*;public class Solution {       /**     * @param n:     n nodes labeled from 0 to n - 1     * @param edges: a undirected graph     * @return: a list of all the MHTs root labels     */    public List
findMinHeightTrees(int n, int[][] edges) { // Wirte your code here List
res = new ArrayList<>(); // 特判一下平凡图,其没有度为1的顶点 if (n == 1) { res.add(0); return res; } // 邻接表建图,并且得到每个顶点的度 Map
> graph = buildGraph(edges); int[] degrees = getDegrees(n, edges); // 将度为1的顶点入队 Queue
queue = new LinkedList<>(); for (int i = 0; i < degrees.length; i++) { if (degrees[i] == 1) { queue.offer(i); } } while (!queue.isEmpty()) { // 把当前层的顶点都加入进答案。等到队列空的时候,最后那一层就是要求的答案 res.clear(); res.addAll(queue); // 由于是分层遍历,所以要记录队列的size int size = queue.size(); for (int i = 0; i < size; i++) { int cur = queue.poll(); for (int next : graph.get(cur)) { degrees[next]--; if (degrees[next] == 1) { queue.offer(next); } } } } return res; } private int[] getDegrees(int n, int[][] edges) { int[] degrees = new int[n]; for (int[] edge : edges) { degrees[edge[0]]++; degrees[edge[1]]++; } return degrees; } private Map
> buildGraph(int[][] edges) { Map
> graph = new HashMap<>(); for (int[] edge : edges) { graph.computeIfAbsent(edge[0], k -> new HashSet<>()).add(edge[1]); graph.computeIfAbsent(edge[1], k -> new HashSet<>()).add(edge[0]); } return graph; }}

时空复杂度 O ( n ) O(n) O(n)

转载地址:http://tgds.baihongyu.com/

你可能感兴趣的文章
mysql中floor函数的作用是什么?
查看>>
MySQL中group by 与 order by 一起使用排序问题
查看>>
mysql中having的用法
查看>>
MySQL中interactive_timeout和wait_timeout的区别
查看>>
mysql中int、bigint、smallint 和 tinyint的区别、char和varchar的区别详细介绍
查看>>
mysql中json_extract的使用方法
查看>>
mysql中json_extract的使用方法
查看>>
mysql中kill掉所有锁表的进程
查看>>
mysql中like % %模糊查询
查看>>
MySql中mvcc学习记录
查看>>
mysql中null和空字符串的区别与问题!
查看>>