代码随想录算法训练营第十七天|110.平衡二叉树，257. 二叉树的所有路径，404.左叶子之和

110.平衡二叉树

题目

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

本题中，一棵高度平衡二叉树定义为：一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过1。

解题思路

题解

• 这题的关键是，只要这个二叉树的任意左右子树的节点高度差大于1，则这个二叉树就已经不是平衡二叉树了。
• 这题仍然使用递归法

代码

class Solution {
private:
    int getHeight(TreeNode *node){
        if(node == nullptr) return 0;
        int leftHeight = getHeight(node->left);
        if (leftHeight == -1) return -1;
        int rightHegiht = getHeight(node->right);
        if (rightHegiht == -1) return -1;
        int result;
        // 如果左子树和右子树的高度差大于1，则不是平衡二叉树，返回-1
        // 否则，当前节点的高度为其左右子节点最大高度，再加上本身的高度1
        if (abs(leftHeight - rightHegiht) > 1) result = -1;
        else result = max(leftHeight, rightHegiht) + 1;
        return result;
    }
public:
    bool isBalanced(TreeNode* root) {
        return getHeight(root) != -1;
    }
};

257. 二叉树的所有路径

题目

给定一个二叉树，返回所有从根节点到叶子节点的路径。

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

解题思路

• 使用前序遍历，只有前序遍历才能从父节点开始指向子节点
• 需要用到回溯的过程，在整个遍历结束之后，需要回退到上一个“岔路口”，走下一条路
  

代码

class Solution {
public:
    vector<string> binaryTreePaths(TreeNode* root) {
        vector<string> result;
        vector<int> path;
        if (root == nullptr) return result;
        traversal(root, path, result);
        return result;
    }

    // path 和 result 传入的参数是地址的copy，不是整个数组
    void traversal(TreeNode *cur, vector<int> &path, vector<string> &result) {

        // 中
        path.push_back(cur->val);

        // 如果左右子节点都为空，说明遍历到了叶子节点，则开始收集path
        if (cur->left == nullptr && cur->right == nullptr) {
            string s;
            for (int i = 0; i < path.size() - 1; ++i) {
                s += to_string(path[i]);
                s += "->";
            }
            // 循环少执行一次，最后加上最后一个数，比如1->2->4，三个数，2个箭头，循环两次后加上最后的数
            s += to_string(path[path.size() - 1]);
            result.push_back(s);
            return;
        }

        // 左
        if (cur->left != nullptr) {
            traversal(cur->left, path, result);
            // 当左节点遍历结束之后，需要将把叶子节点pop出来，然后回溯到上一个节点，再进行下一个遍历
            path.pop_back();
        }

        // 右
        if (cur->right != nullptr) {
            traversal(cur->right, path, result);
            // 同理，遍历有把叶子节点pop出来进行回溯
            path.pop_back();
        }
    }
};

404.左叶子之和

题目

计算给定二叉树的所有左叶子之和。

解题思路

• 这道题说的左叶子节点指的是遍历到最下层之后的左叶子节点
  

代码

这个代码比较繁琐，主要是为了详细的理解后序遍历的过程，以及递归的三步骤。这里主要繁琐在，遍历是遍历到叶子结点的父节点停止的，此时父节点的左子节点即为左叶子结点。

从leetcode中看到了一个比较简单的代码也贴在了下面。

class Solution {
public:
    int sumOfLeftLeaves(TreeNode* root) {
        if (root == nullptr) return 0;
        if (root->left == nullptr && root->right == nullptr) return 0;

        // 左
        int left = sumOfLeftLeaves(root->left);

        // 这里主要是找到叶子节点的同时，保证指针指向的是叶子结点的父节点。
        if (root->left != nullptr && root->left->left == nullptr && root->left->right == nullptr) {
            left = root->left->val;
        }

        // 右
        int right = sumOfLeftLeaves(root->right);

        // 中
        int result = left + right;
        return result;
    }

};

这个版本的代码最巧妙的核心是：

• 每次递归传入一个bool参数来记录当前节点是左节点还是右节点。
• 这样，就不需要记录父节点了，只需要遍历找到左叶子节点即可。
• 初学可以看详细的代码，详细记录了后续遍历过程以及代码整个的思路

class Solution {
public:
    int result;
    // 传入的参数中，is_left 表示当前节点是否为左叶子节点
    void dfs(TreeNode* root, bool is_left) {
    	// 终止条件，如果当前节点为空，则返回
        if(!root) return;
        // 左遍历，并传入true，表示是左节点
        dfs(root->left, true);
        // 右遍历，传入false，表示不是左节点
        dfs(root->right, false);
        // 前两个判断条件是当前节点是否为叶子节点，is_left即可判断当前节点是否为左节点
        if(!root->left && !root->right && is_left) {
            result += root->val;
        }
    }
    int sumOfLeftLeaves(TreeNode* root) {
        result = 0;
        dfs(root, false);
        return result;
    }
};