面试算法题：二叉树的平衡性检测

本文主要是介绍面试算法题：二叉树的平衡性检测，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

更详细的讲解和代码调试演示过程，请参看视频
如何进入google,算法面试技能全面提升指南

如果你对机器学习感兴趣，请参看一下链接：
机器学习：神经网络导论

二叉树由于其结构化清晰，并且变种多样，是面试中极为常见的考题。从本节开始，我们进入到有关二叉树面试题的研究中。

二叉树自身存在着一种递归结构，一个节点除了含有值外，还含有两个节点指针，这两个指针又分别指向两颗二叉树。二叉树有一个很重要的概念叫做树的高，它指的的是从某个节点开始，抵达某个叶子节点的最长路径。例如给定下面的二叉树：

这里写图片描述

如果从根节点6算起，二叉树的高度为4，因为它有4个层级，那么6的左子树，它的高度就为3，以此类推。

如果一颗二叉树是平衡的，必须满足每个节点，它左子树和右子树高度只差不超过1. 问题是，给定一颗二叉树的根节点，给出算法，判断该二叉树是否是一颗平衡二叉树。

如果一棵二叉树是空的，那么我们认为它的高度为0.
对于任意叶子节点，有就是节点的左右子树为空，那么它的高度为1.
对于非叶子节点，它的高度是先计算它的左右子树的高度，那么它本身的高度就是左右子树的最大高度加1.

由此，这道题的解决思路是计算每个节点左右子树的高度，如果两者高度只差大于1，那么它不是平衡的，如果每个节点左右子树的高度只差不超过1，那么他就是一棵平衡二叉树。

二叉树的高度可以递归来计算：
1. 如果输入的是空节点，那么返回高度值0
2. 如果输入的是叶子节点，那么返回高度1
3. 如果输入的是非叶子节点，那么分别计算左右子树的高度，选取其中最大者加1作为本节点的高度。

根据上面思路，我们实现的算法如下：


public class BalancedTree {private boolean balanced = true;public boolean isTreeBalanced(TreeNode node) {computeTreeHeight(node);return balanced;}private int computeTreeHeight(TreeNode node) {if (node == null) {return 0;}int leftHeight = computeTreeHeight(node.left);int rightHeight = computeTreeHeight(node.right);int height = leftHeight > rightHeight ? leftHeight : rightHeight;if (Math.abs(rightHeight - leftHeight) > 1) {balanced = false;}return height + 1;}
}

computeTreeHeight 接收的参数是一个二叉树的节点，然后分别计算该节点的左右子树高度，然后根据结果计算自身高度，在计算过程中，如果发现左右子树高度超过1，那么把balanced 设置成false, 如果该值设置成false的话，那么该二叉树就不是平衡的。

我们看看二叉树节点的定义和构造：


public class TreeNode {public int vaule;public TreeNode left;public TreeNode right;public TreeNode(int v) {this.vaule = v;this.left = this.right = null;}
}public class TreeUtil {private TreeNode root = null;public void addTreeNode(TreeNode node) {if (root == null) {root = node;return;}TreeNode cur = root, prev = root;while (cur != null) {prev = cur;if (cur.vaule > node.vaule) {cur = cur.left;} else {cur = cur.right;}}if (prev.vaule > node.vaule) {prev.left = node;} else {prev.right = node;}}public TreeNode getTreeRoot() {return root;}
}

TreeUtil用来构建一棵二叉树，它构建的是一棵排序二叉树，如果加入的节点比当前节点值小，那么把节点加入当前节点的左子树，如果加入节点的值比当前节点值大，那么把节点加入当前节点的右子树。我们再看看主函数入口处代码：

public class BinaryTree {public static void main(String[] s) {int[] arr = new int[]{6,4,9,2,5,7,10,1,3,8};TreeUtil util = new TreeUtil();for (int i = 0; i < arr.length; i++) {TreeNode n = new TreeNode(arr[i]);util.addTreeNode(n);}TreeNode root = util.getTreeRoot();BalancedTree bt = new BalancedTree();boolean isBalanced = bt.isTreeBalanced(root);System.out.println("If the binary tree is banlanced ? the answer is : " + isBalanced);
public class BinaryTree {public static void main(String[] s) {int[] arr = new int[]{6,4,9,2,5,7,10,1,3,8};TreeUtil util = new TreeUtil();for (int i = 0; i < arr.length; i++) {TreeNode n = new TreeNode(arr[i]);util.addTreeNode(n);}TreeNode root = util.getTreeRoot();BalancedTree bt = new BalancedTree();boolean isBalanced = bt.isTreeBalanced(root);System.out.println("If the binary tree is banlanced ? the answer is : " + isBalanced);util.addTreeNode(new TreeNode(11));util.addTreeNode(new TreeNode(12));util.addTreeNode(new TreeNode(13));root = util.getTreeRoot();isBalanced = bt.isTreeBalanced(root);System.out.println("If the binary tree is banlanced ? the answer is : " + isBalanced);}
}}
}