【程序员面试金典】面试题 04.04. 检查平衡性

题目描述

描述：实现一个函数，检查二叉树是否平衡。在这个问题中，平衡树的定义如下：任意一个节点，其两棵子树的高度差不超过 1。

给定二叉树 [3,9,20,null,null,15,7]
    3
   / \
  9  20
    /  \
   15   7
返回 true 。

给定二叉树 [1,2,2,3,3,null,null,4,4]
      1
     / \
    2   2
   / \
  3   3
 / \
4   4
返回 false 。

解题思路

思路1：最直观的想法是，编写一个dfs函数，实现后序遍历求当前节点高度。检查平衡性即当前节点的左右子树高度差绝对值小于等于1并且左子树平衡并且右子树平衡。由于该题没有告诉节点个数范围，故我没有使用记忆化搜索，其实dfs求高度可以使用记忆化搜索优化时间。

int dfs(TreeNode* cur) //后序遍历求高度
    {
        if(!cur) return 0; //空节点
        int left=dfs(cur->left);  //左
        int right=dfs(cur->right); //右
        return max(left,right)+1; //高度
    }
    bool isBalanced(TreeNode* root) {
        if(!root) return true; //空树
        return abs(dfs(root->left)-dfs(root->right))<=1&&isBalanced(root->left)&&isBalanced(root->right);
    }

思路2：上述是先求高度然后再判断平衡，其实可以边求高度边判断平衡。平衡即左右子树各自平衡，且当前节点的左右子树高度差小于等于1，而后序遍历求高度本身即是后序遍历，其会先遍历左右子树，故可以在遍历左右子树的过程中，判断左右子树的左右子树高度差是否小于等于1，如果不是直接返回-1标记不平衡。

    int dfs(TreeNode* cur) //后序遍历求高度
    {
        if(!cur) return 0; //空节点
        int left=dfs(cur->left);  //左
        if(left==-1)
            return -1;
        int right=dfs(cur->right); //右
        if(right==-1)
            return -1;
        int diff=abs(left-right);
        if(diff>1) return -1;
        return max(left,right)+1; //高度
    }
    bool isBalanced(TreeNode* root) {
        if(!root) return true; //空树
        return dfs(root)==-1?false:true;
    }

总结：先完成，再优化。