問題:
Given a binary tree, determine if it is height-balanced.
For this problem, a height-balanced binary tree is defined as a binary tree in which the depth of the two subtrees of every node never differ by more than 1.
大意:
給出一個二叉樹,判斷它高度是不是平衡的。
對于這個問題,一個高度平衡的二叉樹是指每個節(jié)點的兩個子節(jié)點的深度的差異都不超過1的二叉樹。
思路:
這道題題目說的太簡略了也沒有例子,根據(jù)不斷試錯摸出來的情況,就是說每個節(jié)點的兩個子節(jié)點的深度不能想查大于1,比如左子節(jié)點下還有子節(jié)點,右子節(jié)點下沒有了,這是允許的,深度相差1,但是如果左子節(jié)點的子節(jié)點還有子節(jié)點,深度相差就是2了,就不平衡了。
如果直接判斷有無子節(jié)點,情況太多了不好做。不如直接遞歸記錄每個節(jié)點的深度,然后比較每個節(jié)點的兩個子節(jié)點的深度相差是否大于1。
這里要記錄每個子節(jié)點的深度可以用遞歸的方法,葉子節(jié)點的深度為1,往上遞增,注意在遞歸算一個節(jié)點深度時,要判斷左子節(jié)點和右子節(jié)點的深度哪個更深,取更深的那一個,也就是數(shù)值更大的那一個。
然后可以再判斷每個節(jié)點的左右子節(jié)點的深度相差是否大于1。但是要重新遍歷一遍所有節(jié)點,這無疑增加了時間。在上面我們計算每個節(jié)點的深度的時候,有一步是判斷左右子節(jié)點哪個節(jié)點更深,這里明顯可以直接比較相差是否大于1,如果大于1,我們將該節(jié)點的深度特殊記為-1這個不會出現(xiàn)的深度值,我們希望可以傳遞到最頂層去告訴題目這個二叉樹是不平衡的,因此在計算每個節(jié)點的左右子節(jié)點的時候,需要判斷左右子節(jié)點的深度有沒有等于-1的,如果有,說明下面有個地方出現(xiàn)不平衡了,這時候直接把-1這個信號往上傳就好了。這樣一層層遞歸傳回root,就可以根據(jù)根節(jié)點的深度是否是-1來判斷是否平衡了。
代碼(Java):
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
public class Solution {
public boolean isBalanced(TreeNode root) {
if (height(root) == -1) return false;
else return true;
}
public int height(TreeNode root) {
if (root == null) return 0;
else {
int leftHeight = height(root.left);
if (leftHeight == -1) return -1;
int rightHeight = height(root.right);
if (rightHeight == -1) return -1;
if (leftHeight - rightHeight > 1 || leftHeight - rightHeight < -1) return -1;
return Math.max(height(root.left), height(root.right)) + 1;
}
}
}
合集:https://github.com/Cloudox/LeetCode-Record