題目地址:
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;
}
}