LeetCode-Maximum/Minimum Depth of Binary Tree

我不是女神ヾ 2022-08-04 00:55 136阅读 0赞

iven a binary tree, find its maximum/minimum depth.

The maximum/minimum depth is the number of nodes along the shortest path from the root node down to the nearest leaf node.

两个很相似的题,但是解法完全不怎么一样。

求max比较简单,见如下:

  1. public int maxDepth(TreeNode root) {
  2. if (root == null) return 0;
  3. return Math.max(maxDepth(root.left), maxDepth(root.right))+1;
  4. }

非常精简的代码,彰显了树这种数据结构的特性。

求min就不一样了,因为是root到leaf的距离,所以就必须是叶子节点。如下代码很容易写出来,但是是错了。

  1. public int minDepth(TreeNode root) {
  2. if (root == null) return 0;
  3. return Math.min(minDepth(root.left), minDepth(root.right))+1;
  4. }

因为上述代码并没有保证到叶子节点,为什么求max就可以呢?自己想吧。

更改过之后的版本:

  1. public int minDepth(TreeNode root) {
  2. if (root == null) {
  3. return 0;
  4. }
  5. int left = minDepth(root.left);
  6. int right = minDepth(root.right);
  7. return (left == 0 || right == 0) ? left + right + 1: Math.min(left,right) + 1;
  8. }

但是其实发现,求min值是不需要遍历所有节点的,所以按照层次遍历会节省时间:

  1. public int minDepth(TreeNode root) {
  2. if (root == null) {
  3. return 0;
  4. }
  5. LinkedList<TreeNode> queue = new LinkedList<>();
  6. queue.add(root);
  7. int depth = 0;
  8. while (!queue.isEmpty()) {
  9. depth++;
  10. int len = queue.size();
  11. for (int i = 0; i < len; i++) {
  12. TreeNode node = queue.poll();
  13. if (node.left == null && node.right == null) {
  14. return depth;
  15. }
  16. if (node.left != null) {
  17. queue.add(node.left);
  18. }
  19. if (node.right != null) {
  20. queue.add(node.right);
  21. }
  22. }
  23. }
  24. return 0;
  25. }

发表评论

表情:
评论列表 (有 0 条评论,136人围观)

还没有评论,来说两句吧...

相关阅读