算法训练Day15 | 110.平衡二叉树、257. 二叉树的所有路径、404.左叶子之和

最新推荐文章于 2024-07-20 09:51:47 发布

WIIIter

最新推荐文章于 2024-07-20 09:51:47 发布

阅读量182

点赞数 4

分类专栏： LeetCode刷题文章标签：算法 c++ 数据结构 leetcode

本文链接：https://blog.csdn.net/WIIIter/article/details/136997310

版权

LeetCode刷题专栏收录该内容

49 篇文章 0 订阅

订阅专栏

文章介绍了在LeetCode中的三个与二叉树相关的问题：如何检查平衡二叉树、获取二叉树所有路径以及计算左叶子节点的和。通过递归、层序遍历等方法实现解决方案。

摘要由CSDN通过智能技术生成

110.平衡二叉树

题目链接：110.平衡二叉树-力扣（LeetCode）

思路：递归实现，判断左右子树的的高度差是否超过1，否即为不平衡

class Solution {
public:
    int traversal(TreeNode* node){
        if(!node) return 0;
        int left = traversal(node->left);
        int right = traversal(node->right);
        if(left == -1 || right ==-1 || abs(left-right)>1){
            return -1;
        }else{
            return max(left, right) + 1;
        }
    }
    bool isBalanced(TreeNode* root) {
        if(!root) return true;
        return traversal(root)>=0;

    }
};

参考文章：代码随想录-110.平衡二叉树

257. 二叉树的所有路径

题目链接：257. 二叉树的所有路径-力扣（LeetCode）

思路：递归遍历，记录每一条到达叶子节点的路径，如果传递指针注意回溯

class Solution {
public:
    void traversal(TreeNode* node, vector<int>& path, vector<string>& result){
        path.push_back(node->val);
        if(!node->left && !node->right){
            string temp;
            for(int i=0; i<path.size()-1; i++){
                temp += to_string(path[i]);
                temp += "->";
            }
            temp += to_string(path[path.size()-1]);
            result.push_back(temp);
            return;
        }
        if(node->left){
            traversal(node->left, path, result);
            path.pop_back();
        }
        if(node->right){
            traversal(node->right, path, result);
            path.pop_back();
        }
    }
    vector<string> binaryTreePaths(TreeNode* root) {
        vector<int> path;
        vector<string> result;
        if(!root) return result;
        traversal(root, path, result);
        return result;

    }
};

参考文章：代码随想录-257. 二叉树的所有路径

404.左叶子之和

题目链接：404.左叶子之和-力扣（LeetCode）

思路：层序遍历，记录每一层的第一个叶子节点，求和

class Solution {
public:
    int traversal(TreeNode* node, bool flag){
        if(!node->left && !node->right){
            if(flag) return node->val;
            else return 0;
        }
        int num = 0;
        if(node->left) num += traversal(node->left, true);
        if(node->right) num += traversal(node->right, false);

        return num;

    }
    int sumOfLeftLeaves(TreeNode* root) {
        int num = 0;
        if(root->left){
            num+=traversal(root->left, true);
        }
        if(root->right){
            num+=traversal(root->right, false);
        }
        return num;

    }
};