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

目录

题目链接：110.平衡二叉树

思路

代码

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

思路

步骤

代码

题目链接：404.左叶子之和

思路

代码

总结

---

题目链接：110.平衡二叉树

思路

        平衡二叉树的定义是所有节点的左右子树的高度差不大于1，所以只需要判断每个节点的左右子树的高度差即可。求高度，需要从叶子节点向根节点遍历，所以这里采用后序遍历，左右中，先获取孩子节点的信息，再返回给父节点，得到高度。

代码

class Solution {
public:
    int getHeight(TreeNode* root) {
        if (root == NULL)
            return 0; // 空节点时返回高度0
        // 后序遍历
        int lefth = getHeight(root->left); // 左
        if (lefth == -1)
            return -1;
        int righth = getHeight(root->right); // 右
        if (righth == -1)
            return -1;
        int result;
        if (abs(lefth - righth) > 1)
            result = -1;
        else {
            result = 1 + max(lefth, righth); // 中
        }
        return result;
    }
    bool isBalanced(TreeNode* root) {
        if (getHeight(root) == -1)
            return false;
        else
            return true;
    }
};

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

思路

        路径都是从根节点指向叶子节点的，因此要选择前序遍历，中左右，先将中间节点存进去，然后再去向下遍历。同时，终止条件是到最后的叶子节点就返回路径，而不是到空节点。

        回溯过程，要存储所有路径，根节点或者说中间节点就要被重复使用，每条路径探索完之后，要将这条路径返回到中间节点，再去探索别的路径，从叶子节点往根节点回退的过程就是回溯过程。

步骤

        ①将中间节点存到路径的第一个位置

        ②判断当前这个中间节点是否是最后一个节点，如果是，返回路径

        ③向左遍历

        ④向右遍历

代码

class Solution {
public:
    // 返回类型为空，这里是对参数中的数组进行处理
    // 第一个参数是传入的根节点
    // 第二个参数是存放的路径，数字
    // 第三个参数是结果数组，需要将int型数组转换成string，并且加上->符号
    void reversal(TreeNode* root, vector<int>& path, vector<string>& result) {
        // 前序遍历，中左右
        path.push_back(root->val); // 中，传入中间节点的值
        if (root->left == NULL && root->right == NULL) {
            string tmp; // 定义临时字符串用来保存当前路径
            for (int i = 0; i < path.size() - 1; i++) {
                tmp +=
                    to_string(path[i]); // 存入当前数值，并将其转换成string类型
                tmp += "->";
            }
            // 这里单独对最后一个数值进行处理
            // 不然在结果的末尾会多出一个"->"
            tmp += to_string(path[path.size() - 1]);
            result.push_back(tmp); // 将保存的当前路径存到结果数组中
            return;
        }
        if (root->left) {
            reversal(root->left, path, result); // 左
            path.pop_back();                    // 回溯
        }
        if (root->right) {
            reversal(root->right, path, result); // 右
            path.pop_back();
        }
    }
    vector<string> binaryTreePaths(TreeNode* root) {
        vector<int> path;
        vector<string> result;
        reversal(root, path, result);
        return result;
    }
};

题目链接：404.左叶子之和

思路

        求所有左叶子之和，首先要判断是否是叶子节点，即该节点的左右孩子为空，其次要判断该节点是否是其父节点的左孩子，因此在遍历到其父节点时，就要对左孩子进行判断。这时已经确定了它是左孩子，只需要判断它是否是叶子节点即可。

代码

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); // 右
        int sum = leftnum + rightnum;                // 中
        return sum;
    }
};

总结

        ①虽然已经做了二叉树的一些题，并且了解了前中后序遍历，层序遍历，但是遇到新题的时候还是没有清晰的思路

        ②首先明确题目要做什么，其次选择遍历方式，最后明确处理方式

        ③递归三部曲很重要

        ④二叉树的细化处理还是要加强，之前都是对整个二叉树进行遍历，然后采用相同的处理方式。这种对特定节点，路径以及二叉树的类型判断等问题还要多练