LeetCode 热题 C++ 102. 二叉树的层序遍历 104. 二叉树的最大深度

最新推荐文章于 2023-03-20 12:04:05 发布

Zero_979

最新推荐文章于 2023-03-20 12:04:05 发布

阅读量110

点赞数

分类专栏：力扣树文章标签： c++ leetcode 算法

本文链接：https://blog.csdn.net/zero_979/article/details/128367194

版权

力扣同时被 2 个专栏收录

52 篇文章 0 订阅

订阅专栏

树

20 篇文章 0 订阅

订阅专栏

力扣102

给你二叉树的根节点 root ，返回其节点值的 层序遍历 。（即逐层地，从左到右访问所有节点）。

示例 1：

输入：root = [3,9,20,null,null,15,7]
输出：[[3],[9,20],[15,7]]

示例 2：

输入：root = [1]
输出：[[1]]

示例 3：

输入：root = []
输出：[]

提示：

树中节点数目在范围 [0, 2000] 内
-1000 <= Node.val <= 1000

思路：

层序遍历是简单的，只要用到队列就可以。但关键是怎么判断每一层结束。

因为每一层都是当前队列中元素的孩子，所以在清空当前队列的时候，把它们的孩子放进vector里就行啦。

代码：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    vector<vector<int>> levelOrder(TreeNode* root) {
        vector<vector<int> >v;
        vector<int> vv;
        if(root==nullptr)return v;
        queue<TreeNode*> q;
        q.push(root);
        vv.push_back(root->val);
        //int k=0;
        TreeNode* p;
        while(q.size()){
            v.push_back(vv);
            vv.clear();
            int k=q.size();
            for(int i=0;i<k;i++)
            {
                if(!q.size())break;
                p=q.front();
                q.pop();
                if(p->left){
                    vv.push_back(p->left->val);
                    q.push(p->left);
                }
                if(p->right){
                    vv.push_back(p->right->val);
                    q.push(p->right);
                }
            }
            k++;
        }
        if(vv.size())v.push_back(vv);
        return v;
    }
};

力扣104

给定一个二叉树，找出其最大深度。

二叉树的深度为根节点到最远叶子节点的最长路径上的节点数。

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

示例：
给定二叉树 [3,9,20,null,null,15,7]，

返回它的最大深度 3 。

思路：

和上一题一模一样，就删减几句就好了。

代码：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    int maxDepth(TreeNode* root) {
        if(root==nullptr)return 0;
        queue<TreeNode*> q;
        q.push(root);
        TreeNode* p;
        int ans=0;
        while(q.size()){
            int k=q.size();
            for(int i=0;i<k;i++)
            {
                if(!q.size())break;
                p=q.front();
                q.pop();
                if(p->left)q.push(p->left);
                if(p->right)q.push(p->right);
            }
            ans++;
        }
        return ans;
    }
};