力扣(LeetCode)543

Linus脫襪子發表於2019-01-19

題目地址:
https://leetcode-cn.com/probl…
題目描述:
給定一棵二叉樹,你需要計算它的直徑長度。一棵二叉樹的直徑長度是任意兩個結點路徑長度中的最大值。這條路徑可能穿過根結點。

示例 :
給定二叉樹

      1
     / 
    2   3
   /      
  4   5    

返回 3, 它的長度是路徑 [4,2,1,3] 或者 [5,2,1,3]。

注意:兩結點之間的路徑長度是以它們之間邊的數目表示。

解答:
這一題就是找到每個節點的深度(等於max(左節點深度,右節點深度)+1),因為穿過任意一個節點的
直徑,為這個節點左子樹的深度+右子樹的深度。因此只需要計算出每個節點深度,就可以得到穿過
某節點的直徑。因此得到深度之後可以選擇利用dfs深度優先搜尋來找到答案,也可以在求深度的時候
就直接計算,然後儲存在hashmap中。
java ac程式碼:

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public int diameterOfBinaryTree(TreeNode root) {
        if(root == null)return 0;
        HashMap<TreeNode,Integer> map = new HashMap(500);
        height(root,map);
        int ans = 0;
        for(Map.Entry<TreeNode,Integer> entry:map.entrySet())
            ans = Math.max(ans,entry.getValue());
        return ans;
        
    }
    public int height(TreeNode root,HashMap<TreeNode,Integer>map)
    {
        if(root == null)return 0;
        int left = height(root.left,map);
        int right = height(root.right,map);
        int ans = left + right;
        map.put(root,ans);
        return Math.max(left,right)+1;
    }
}


相關文章