码迷,mamicode.com
首页 > 其他好文 > 详细

Leetcode104 二叉树最大深度 踩坑

时间:2020-02-12 23:48:08      阅读:90      评论:0      收藏:0      [点我收藏+]

标签:code   难度   fir   max   nbsp   提交代码   push   复习   tco   

104题难度为简单,可本人却画了不少时间,现将遇到的坑写在此,方便日后复习。大家应该不会遇到这么蠢的问题。

1.递归方法很简单,也很巧妙这里就不说了。

2.下面主要是深度遍历方法中本人遇到的坑。

3.看到题目想到可以通过深度遍历来解决问题,借助栈把深度计算出来(坑就在这个栈里,这种题目还是先画个图看看,再撸代码吧,笑哭了)。

下面是本人的代码,思想:将每层元素存入栈中,然后再逐一弹出,并深度标志加1。大概想了一下应该问题不大,题目给的实列也通过了,可提交代码时问题来了

不能通过所有实例。并给出了没通过的树,如下:[1,2,3,4,null,null,5]; 

 1 /**
 2  * Definition for a binary tree node.
 3  * public class TreeNode {
 4  *     int val;
 5  *     TreeNode left;
 6  *     TreeNode right;
 7  *     TreeNode(int x) { val = x; }
 8  * }
 9  */
10 class Solution {
11     public int maxDepth(TreeNode root) {
12         if(root == null){
13             return 0;
14         }
15         Stack<TreeNode> stack = new Stack<>();
16         int num = 0;
17         stack.push(root);
18         while(!stack.isEmpty()){
19             num++;
20             int size = stack.size();
21             for(int i = 0;i < size;i++){
22                 TreeNode cur = stack.pop();
23                 if(cur.left != null)
24                     stack.push(cur.left);
25                 if(cur.right != null)
26                     stack.push(cur.right);
27             }
28         }
29         return num;
30     }
31 }

 

此树的最大深度为3,而本人代码输出为4。画图后自己推一下,确实是4。因为栈总是先弹出后进的元素当,代码运行到元素3时,应该接着弹出2才对,可栈会弹出5,那么问题就来了,

当5弹出后此时深度标志就已经为3.当下次循环弹出元素2时,左子树还存在元素4。导致多一次循环,输出为4。

4.通过观察发现如果符合先进先出就可符合题意,当然想到了LinkedList。add添加元素,pollFirst弹出先进如元素。完美解决问题。代码如下:

 1 /**
 2  * Definition for a binary tree node.
 3  * public class TreeNode {
 4  *     int val;
 5  *     TreeNode left;
 6  *     TreeNode right;
 7  *     TreeNode(int x) { val = x; }
 8  * }
 9  */
10 class Solution {
11     public int maxDepth(TreeNode root) {
12         if(root == null){
13             return 0;
14         }
15         LinkedList<TreeNode> stack = new LinkedList<>();
16         int num = 0;
17         stack.add(root);
18         while(!stack.isEmpty()){
19             num++;
20             int size = stack.size();
21             for(int i = 0;i < size;i++){
22                 TreeNode cur = stack.pollFirst();
23                 if(cur.left != null){
24                     stack.add(cur.left);
25                 }
26                 if(cur.right != null){
27                     stack.add(cur.right);
28                 }  
29             }
30         }
31         return num;
32     }
33 }

 

5.主要没有考虑完善栈先进后出的特性。

记递归代码如下:

 1 /**
 2  * Definition for a binary tree node.
 3  * public class TreeNode {
 4  *     int val;
 5  *     TreeNode left;
 6  *     TreeNode right;
 7  *     TreeNode(int x) { val = x; }
 8  * }
 9  */
10 class Solution {
11     public int maxDepth(TreeNode root) {
12         if(root == null){
13             return 0;
14         }else{
15             int i = maxDepth(root.left);
16             int j = maxDepth(root.right);
17             return i = (i>j?i:j) + 1;
18         } 
19     }
20 }

 

Leetcode104 二叉树最大深度 踩坑

标签:code   难度   fir   max   nbsp   提交代码   push   复习   tco   

原文地址:https://www.cnblogs.com/Guo-xin/p/12301842.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!