代码随想录算法训练营第十七天 | LeetCode110 平衡二叉树 LeetCode257 二叉树所有的路径 LeetCode404 左叶子之和

LeetCode110 平衡二叉树

题目链接：平衡二叉树

文章链接：平衡二叉树

视频链接：平衡二叉树

思路

一棵高度平衡二叉树定义为：一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过1。高度是指距离叶子节点的距离，深度是只距离根节点的距离。**求高度用后序遍历，求深度用前序遍历。**整体思路是求出左右子树的高度之差，并返回给父节点，判断该父节点是否是平衡二叉树，若所有节点都满足平衡二叉树，则该二叉树为平衡二叉树。

代码

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;
    }
};

LeetCode257 二叉树所有的路径

题目链接：二叉树所有的路径

文章链接：二叉树所有的路径

视频链接：二叉树所有的路径

思路

这题涉及到回溯，有点不太熟悉，直接看视频和文字讲解了。

代码

class Solution {
private:

    void traversal(TreeNode* cur, vector<int>& path, vector<string>& result) {
        path.push_back(cur->val); // 中，中为什么写在这里，因为最后一个节点也要加入到path中 
        // 这才到了叶子节点
        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(); // 回溯
        }
    }

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

LeetCode404 左叶子之和

题目链接：左叶子之和

文章链接：左叶子之和

视频链接：左叶子之和

思路

首先得明白左叶子的定义，即节点A的左孩子不为空，且左孩子的左右孩子都为空（说明是叶子节点），那么A节点的左孩子为左叶子节点，而不能通过当前节点判断是否为左叶子，必须通过当前节点的父节点来判断，如果父节点的左节点的左右节点为空，则该父节点的左节点为左叶子。因为需要通过父节点来判断，则选择后序遍历，因为后序遍历的顺序是左右中，刚好左右节点判断完后将结果返回给父节点。注意，只有当前遍历的节点是父节点，才能判断其子节点是不是左叶子。 所以如果当前遍历的节点是叶子节点，那其左叶子也必定是0。

代码

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

        int leftValue = sumOfLeftLeaves(root->left);    // 左
        if (root->left && !root->left->left && !root->left->right) { // 左子树就是一个左叶子的情况
            leftValue = root->left->val;
        }
        int rightValue = sumOfLeftLeaves(root->right);  // 右

        int sum = leftValue + rightValue;               // 中
        return sum;
    }
};