【算法训练 day20 平衡二叉树、二叉树的路径、左叶子之和】

---

一、平衡二叉树-LeetCode 110

Leecode链接: leetcode 977
文章链接: 代码随想录
视频链接: B站

给定一个二叉树，判断它是否是
平衡二叉树

示例：

输入：root = [3,9,20,null,null,15,7]
输出：true

思路

判断是否是平衡的，需要根据左右子树的高度差来判断，因为涉及到高度，所以函数返回值需要int型，在遇到左右子树高度差大于1时该返回-1，表示不平衡。

实现代码

//cpp
class Solution {
public:
    // 返回以该节点为根节点的二叉树的高度，如果不是平衡二叉树了则返回-1
    int getHeight(TreeNode* node) {
        if (node == NULL) {
            return 0;
        }
        int leftHeight = getHeight(node->left);
        if (leftHeight == -1) return -1;
        int rightHeight = getHeight(node->right);
        if (rightHeight == -1) return -1;
        return abs(leftHeight - rightHeight) > 1 ? -1 : 1 + max(leftHeight, rightHeight);
    }
    bool isBalanced(TreeNode* root) {
        return getHeight(root) == -1 ? false : true;
    }
};

个人问题

有思路没实现代码，个人觉得涉及高度问题，左右子树必须返回高度值，这样才能判断是否平衡，但不平衡时该返回什么，当时没有想清楚，只觉得应该返回bool值，但返回bool值怎么判断高度差，最终导致代码没有写出来。

---

二、二叉树的路径-LeetCode 257

Leecode链接: LeetCode 257
文章链接: 代码随想录
视频链接: B站

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

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

示例：

输入：root = [1,2,3,null,5]
输出：["1->2->5","1->3"]

思路

使用回溯加递归，遍历时使用一个path数组记录遍历到的数字，遍历到叶子节点就插入到res数组中记录下来，然后回退path。

实现代码

//cpp
class Solution {
private:
    void traversal(TreeNode* cur, string path, vector<string>& result) {
        path += to_string(cur->val); // 中，中为什么写在这里，因为最后一个节点也要加入到path中
        if (cur->left == NULL && cur->right == NULL) {
            result.push_back(path);
            return;
        }
        if (cur->left) {
            path += "->";
            traversal(cur->left, path, result); // 左
            path.pop_back(); // 回溯 '>'
            path.pop_back(); // 回溯 '-'
        }
        if (cur->right) {
            path += "->";
            traversal(cur->right, path, result); // 右
            path.pop_back(); // 回溯'>'
            path.pop_back(); // 回溯 '-'
        }
    }

public:
    vector<string> binaryTreePaths(TreeNode* root) {
        vector<string> result;
        string path;
        if (root == NULL) return result;
        traversal(root, path, result);
        return result;

    }
};

个人问题

有思路但没写出代码。

总结

题目需要注意插入节点值这个操作需要放最上面，因为退出条件是遍历到叶子节点就插入res数组，这样可以避免没有插入path数组就直接将结果放到了res数组中；值拷贝的原因在于回溯思想，这样可以避免来回对path操作。

---

三.左叶子之和-LeeCode 404

Leecode链接: LeetCode 404
文章链接: 代码随想录
视频链接: B站

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

示例：

输入: root = [3,9,20,null,null,15,7] 
输出: 24 
解释: 在这个二叉树中，有两个左叶子，分别是 9 和 15，所以返回 24

思路

使用后序遍历，计算各个子树中的左叶子之和，然后相加。

实现代码

//cpp
class Solution {
public:
    int sumOfLeftLeaves(TreeNode* root) {
        if(root == nullptr){
            return 0;
        }

        int leftsum = sumOfLeftLeaves(root->left);
        if(root->left && root->left->left == nullptr && root->left->right == nullptr){
            leftsum = root->left->val;
        }
        int rightsum = sumOfLeftLeaves(root->right);

        int sum = leftsum + rightsum;
        return sum;
    }
};

个人问题

代码没有自己写出来。

总结

需要注意的是，本体不仅需要判断是不是叶子节点还要判断是不是左叶子节点，所以不能用当前节点来判断，而需要借助父节点来判断。时间复杂度O(n^2),空间复杂度O(1)。