104. 二叉树的最大最小深度以及平衡二叉树

104. 二叉树的最大深度

给定一个二叉树，找出其最大深度。

二叉树的深度为根节点到最远叶子节点的最长路径上的节点数。

说明: 叶子节点是指没有子节点的节点。

思路

直接DFS，返回左右子树深度大的即可。

代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    int maxDepth(TreeNode* root) {
        if(root == nullptr) {
            return 0;
        }
        return max(maxDepth(root -> left), maxDepth(root -> right)) + 1;
    }

};

111. 二叉树的最小深度

给定一个二叉树，找出其最小深度。

最小深度是从根节点到最近叶子节点的最短路径上的节点数量。

说明: 叶子节点是指没有子节点的节点。

思路

单纯的我以为直接返回左右子树最小的即可，结果样列[1, 2]输出1，正确答案是2。
正确的递归结束条件：

• 左右子树均为空，返回1
• 左子树为空，返回右子树深度，右子树为空，返回左子树深度
• 左右子树均不为空，返回左右子树最小深度

代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    int minDepth(TreeNode* root) {
        if(root == nullptr) {
            return 0;
        }
        int leftDepth = minDepth(root -> left);
        int rightDepth = minDepth(root -> right);
		//为空的树深度为0,直接可以相加返回
        if(root -> left == nullptr || root -> right == nullptr) {
            return leftDepth + rightDepth + 1;
        }
        
        return 1 + min(leftDepth, rightDepth);
    }
};

110. 平衡二叉树

给定一个二叉树，判断它是否是高度平衡的二叉树。

本题中，一棵高度平衡二叉树定义为：

一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过1。

思路

想到了二叉树的最大深度的解法，利用最大深度求法，分别求得节点的左右子树的最大深度，比较大小即可，然后递归判断左右子树是否又是平衡二叉树。

代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    bool isBalanced(TreeNode* root) {
        if(root == nullptr) {
            return true;
        }
        return abs(dfs(root -> left) - dfs(root -> right)) <= 1 && isBalanced(root -> left) && isBalanced(root -> right); 
    }
    int dfs(TreeNode* root) {
        if(root == nullptr) {
            return 0;
        }
        return max(dfs(root -> left), dfs(root -> right)) + 1;
    }
};