劍指offer - 55 - I. 二叉樹的深度

題目描述

輸入一棵二叉樹的根節點，求該樹的深度。從根節點到葉節點依次經過的節點（含根、葉節點）形成樹的一條路徑，最長路徑的長度爲樹的深度。

例如：

給定二叉樹 [3,9,20,null,null,15,7]，

3

/
9 20
/
15 7
返回它的最大深度 3 。

提示：

節點總數 <= 10000
注意：本題與leetcode 104 題相同。

思路1：先序遍歷

二叉樹的深度 = 左右子樹深度的較大值 + 1

代碼

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

執行用時 :
0 ms, 在所有 Java 提交中擊敗了100.00%的用戶
內存消耗 :
39.6 MB, 在所有 Java 提交中擊敗了100.00%的用戶

思路2：層序遍歷

用隊列實現。每遍歷一層，則計數器 +1 ，直到遍歷完成，則可得到樹的深度。

class Solution {
    public int maxDepth(TreeNode root) {
        if(root == null) return 0;
        Queue<TreeNode> queue = new LinkedList<>() {{ add(root); }};
        int res = 0;
        while(!queue.isEmpty()) {
        	for(int i = queue.size(); i >0 ; i--) {
        		TreeNode current = queue.poll();
        		if (current.left != null) queue.add(current.left);
        		if (current.right != null) queue.add(current.right);
        	}
            res++;
        }
        return res;
    }
}