LeetCode-110-平衡二叉樹

平衡二叉樹

題目描述：給定一個二叉樹，判斷它是否是高度平衡的二叉樹。
本題中，一棵高度平衡二叉樹定義為：
一個二叉樹每個節點的左右兩個子樹的高度差的絕對值不超過 1 。
示例說明請見LeetCode官網。
來源：力扣（LeetCode）
連結：https://leetcode-cn.com/probl...
著作權歸領釦網路所有。商業轉載請聯絡官方授權，非商業轉載請註明出處。

解法一：遞迴

首先，新增一個方法height，該方法通過層序遍歷的方式得到二叉樹的高度。
然後，通過遞迴法判斷二叉樹是否是平衡二叉樹，遞迴過程如下：
如果當前根節點為空，則直接返回true；
否則，計算當前根節點的左右子樹的高度，如果當前根節點的左右子樹的高度之差不超過1，則遞迴判斷當前根節點的左右子樹是否是平衡二叉樹；否則，返回false。

import com.kaesar.leetcode.TreeNode;

import java.util.LinkedList;
import java.util.Queue;

public class LeetCode_110 {
    /**
     * 遞迴
     *
     * @param root
     * @return
     */
    public static boolean isBalanced(TreeNode root) {
        // 如果當前根節點為null，肯定是平衡的，直接返回true
        if (root == null) {
            return true;
        }
        // 如果當前根節點的左右子樹的高度之差不超過1，則遞迴判斷當前根節點的左右子樹是否是平衡二叉樹
        if (height(root.left) - height(root.right) >= -1 && height(root.left) - height(root.right) <= 1) {
            return isBalanced(root.left) && isBalanced(root.right);
        } else {
            return false;
        }
    }

    /**
     * 層序遍歷：通過計算二叉樹的層數來得到二叉樹的高度
     *
     * @param root 當前根節點
     * @return 返回二叉樹的高度
     */
    public static int height(TreeNode root) {
        int result = 0;
        if (root == null) {
            return result;
        }
        Queue<TreeNode> nodes = new LinkedList<>();
        nodes.add(root);
        result = 0;
        while (!nodes.isEmpty()) {
            result++;
            int count = nodes.size();
            while (count > 0) {
                TreeNode cur = nodes.poll();
                if (cur.left != null) {
                    nodes.add(cur.left);
                }
                if (cur.right != null) {
                    nodes.add(cur.right);
                }
                count--;
            }
        }
        return result;
    }

    public static void main(String[] args) {
        // 測試用例，期望返回結果： true
        TreeNode root = new TreeNode(3);
        root.left = new TreeNode(9);
        root.right = new TreeNode(20);
        root.right.left = new TreeNode(15);
        root.right.right = new TreeNode(7);

        System.out.println(isBalanced(root));
    }
}

【每日寄語】 把所有的不快給昨天，把所有的希望給明天，把所有的努力給今天。

LeetCode-110-平衡二叉樹

平衡二叉樹

解法一：遞迴

相關文章