LeetCode：面试题 04.04. 检查平衡性

最新推荐文章于 2024-02-10 07:45:00 发布

大秦隐士

最新推荐文章于 2024-02-10 07:45:00 发布

阅读量196

点赞数

分类专栏： LeetCode 文章标签： leetcode

本文链接：https://blog.csdn.net/daqin_yinshi/article/details/107110783

版权

LeetCode 专栏收录该内容

45 篇文章 0 订阅

订阅专栏

深度优先搜索树，得到左子树高度和右子树高度，如果两个高度差超过1，则不是平衡树

public class MyTest1 {

	private boolean isBalance = true;
	public boolean isBalanced(TreeNode root) {
		dfs(root);
		return isBalance;
	}
	
	/**
	 * 深度优先，遍历树，返回树的高度
	 * @param root
	 * @return
	 */
	private int dfs(TreeNode root) {
		if(root==null) {
			return 0;
		}
		
		int leftDepth = dfs(root.left);
		int rightDepth = dfs(root.right);
		
		if(Math.abs(rightDepth-leftDepth)>1) {
			isBalance = false;
		}
		
		return Math.max(leftDepth, rightDepth)+1;
	}
}