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

最新推荐文章于 2024-10-02 22:45:05 发布

小刘很ok

最新推荐文章于 2024-10-02 22:45:05 发布

阅读量157

点赞数 1

分类专栏： leetcode刷题打卡文章标签：算法 leetcode 数据结构

本文链接：https://blog.csdn.net/weixin_44047621/article/details/127814143

版权

leetcode刷题打卡专栏收录该内容

52 篇文章 3 订阅

订阅专栏

110.平衡二叉树

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

此题与前面求深度的类似，但是也有不同之处，注意不满足平衡二叉树条件时返回-1标记，等最后判断如果是-1就返回false；
代码可以简洁，但是隐藏了很多细节，初期应该尽量体现细节。

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

257.二叉树的所有路径

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

此题第一次出现回溯，代码写的详细点可以体现回溯。只有回溯才能将所有情况考虑到。

采用前序遍历方法

class Solution {
public:
    void traversal(TreeNode* cur, vector<int>& path, vector<string>& result) {
        path.push_back(cur->val); //中
        if (cur->left == nullptr && cur->right == nullptr) {
            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<string> result;
        vector<int> path;
        if (root == nullptr) return result;
        traversal(root, path, result);
        return result;
    }
};

404.左叶子之和

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

注意左叶子首先是叶子节点，其次他是父节点的左孩子。

采用后序遍历，不断统计左右子树左叶子节点和。

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

        int leftNum = sumOfLeftLeaves(root->left);
        //左叶子的值
        if (root->left != nullptr && root->left->left == nullptr && root->left->right == nullptr) {
            leftNum = root->left->val;
        }
        int rightNum = sumOfLeftLeaves(root->right);
        //统计每个节点左右孩子中左叶子的和
        int sum = leftNum + rightNum;
        return sum;
    }
};