【代码随想录】算法训练营 第十七天 第六章 二叉树 Part 4

110. 平衡二叉树

题目

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

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

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

 

思路

只要这个二叉树里任何一个结点的左右子树不平衡，这个二叉树就不是平衡二叉树，而平衡则意味着左右子树的高度差绝对值不超过1，所以我们应该先弄清楚什么是高度。

我们可以想象一个根节点在上，下面有子树的二叉树，选定一个节点，从根节点往下数到这个节点的数值就是深度，而从这个节点往下数到叶子节点的数值就是高度，简单来说，从上往下是深度，从下往上是高度，根节点的深度一定为1，叶子节点的高度一定为1.

搞懂了什么是高度，接下来做题只需要从下往上返回就可以了，这就需要用到二叉树的后序遍历，每次先遍历左右子树，最后返回到它们的根节点做判断，遍历左右子树会得到它们的深度，如果它们不平衡，则记深度为-1，在遍历后就可以判断，如果两个都不等于-1，就可以返回到根节点判断高度差的绝对值，绝对值大于1就是不平衡，往上返回-1。

代码

class Solution {
public:
    int getHeight(TreeNode* node) {
        if (node == NULL) return 0;
        int leftHeight = getHeight(node->left);
        int rightHeight = getHeight(node->right);
        if (leftHeight == -1 || rightHeight == -1) return -1;
        return abs(leftHeight - rightHeight) > 1 ? -1 : 1 + max(leftHeight, rightHeight);
    }
    bool isBalanced(TreeNode* root) {
        return getHeight(root) == -1 ? false : true;
    }
};

257. 二叉树的所有路径

题目

给你一个二叉树的根节点 root ，按 任意顺序 ，返回所有从根节点到叶子节点的路径。

叶子节点 是指没有子节点的节点。

思路

一个递归走天下！还是递归求解，从根节点开始向下走，所以这里用的是前序遍历，我们要递归得到的是所有从根节点到叶子节点的路径，所以按照递归三部曲，首先要确定的是参数，那就是当前的二叉树结点、记录路径的vector容器以及保存答案字符串的vector容器；其次是递归终止条件，我们每次向下走，走到叶子节点就可以停下来保存路径了，所以终止条件必然是遇到了叶子节点，遇到了以后，我们设置一个字符串变量来保存这次的路径；接着就要分别遍历左子树和右子树了，左右子树每次递归遍历以后，都要把路径中的最后一个节点弹出，这样才能遍历完。

盲区

在保存字符串的时候，用到了一个变量类型转化的函数 to_string：

sPath += to_string(path[i]);

而且，在字符串后添加可以直接用加法，学到了。

代码

class Solution {
public:
    void traversal(TreeNode* cur, vector<int> &path, vector<string> &result) {
        path.push_back(cur->val);
        if (cur->left == NULL && cur->right == NULL) {
            string sPath;
            for (int i = 0; i < path.size() - 1; i++) {
                sPath += to_string(path[i]);
                sPath += "->";
            }
            sPath += to_string(path[path.size() - 1]);
            result.push_back(sPath);
            return;
        }
        if (cur->left) {
            traversal(cur->left, path, result);
            path.pop_back();
        }
        if (cur->right) {
            traversal(cur->right, path, result);
            path.pop_back();
        }
    }
    vector<string> binaryTreePaths(TreeNode* root) {
        vector<int> path;
        vector<string> result;
        if (root == NULL) return result;
        traversal(root, path, result);
        return result;
    }
};

404. 左叶子之和

题目

给定二叉树的根节点 root ，返回所有左叶子之和。

思路

刚开始自己尝试的时候摔沟里了，以为是所有左节点之和，做了半天没做对，最后乖乖看了讲解视频。

按照随想录的思路，我们要统计左叶子之和，还是要用递归的方法来，首先明确参数，这里的参数其实就是二叉树结点，返回值是当前的左叶子之和，接下来看终止条件，我们递归到一个空节点的时候就返回0，如果递归到叶子节点也是，因为我们把左叶子的值加入答案并不是递归到叶子节点的时候做的，而是在其父节点的时候就得判断，所以叶子节点也是返回0；下面就要分别递归遍历左子树和右子树，右子树不用多加，左子树就得在判断后返回左叶子的值，最后把左右两边统计到的值返回上去就行。

代码

class Solution {
public:
    int sumOfLeftLeaves(TreeNode* root) {
        if (root == NULL) return 0;
        if (root->left == NULL && root->right == NULL) return 0;
        int leftnum = sumOfLeftLeaves(root->left);
        if (root->left != NULL && root->left->left == NULL && root->left->right == NULL) leftnum = root->left->val;
        int rightnum = sumOfLeftLeaves(root->right);
        return leftnum + rightnum;
    }
};