LeetCode——二叉树的层序遍历

简介: LeetCode——二叉树的层序遍历

102. 二叉树的层序遍历 I

给你二叉树的根节点 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

原题链接:https://leetcode.cn/problems/binary-tree-level-order-traversal/

思路

这个只要用一个队列帮助就可以了:

先存储第一个结点。

然后访问第一个结点的左右子树,第一个结点出队。

区分层数采用vector<vector< int>>的方式,用循环,里面的vector< int>是一层,放入一层二叉树的数据,然后再次进入就是下一层了。

代码

class Solution {
public:
    vector<vector<int>> levelOrder(TreeNode* root) {
        vector<vector<int>>arr;
        queue<TreeNode*>q;
        int qsize = 0;//记录结点数用的
        if(root)
        {
            q.push(root);//入队
            qsize = 1;//结点数1
        }
        while(!q.empty())
        {
            vector<int>ar;//一层结点的数据储存
            while(qsize--)//走完层数和结点也有关
            {
                TreeNode*p = q.front();
                q.pop();//出队
                ar.push_back(p->val);//让数据进入
                if(p->left)//不为空入队
                    q.push(p->left);
                if(p->right)
                    q.push(p->right);
            }
            arr.push_back(ar);//将这一层的数据放入arr中
            qsize = q.size();//下一层的结点个数
        }
        return arr;
    }
};

107. 二叉树的层序遍历 II

给你二叉树的根节点 root ,返回其节点值 自底向上的层序遍历 。 (即按从叶子节点所在层到根节点所在的层,逐层从左向右遍历)

示例 1

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

输出:[[15,7],[9,20],[3]]

示例 2

输入:root = [1]

输出:[[1]]

示例 3

输入:root = []

输出:[]

提示

树中节点数目在范围 [0, 2000] 内

-1000 <= Node.val <= 1000

原题链接:https://leetcode.cn/problems/binary-tree-level-order-traversal-ii/

思路

这道题只要将上面数组反转一下就可以了。

代码

class Solution {
public:
    vector<vector<int>> levelOrderBottom(TreeNode* root) {
        vector<vector<int>>arr;
        queue<TreeNode*>q;
        int qsize = 0;//记录结点数用的
        if(root)
        {
            q.push(root);//入队
            qsize = 1;//结点数1
        }
        while(!q.empty())
        {
            vector<int>ar;//一层结点的数据储存
            while(qsize--)//走完层数和结点也有关
            {
                TreeNode*p = q.front();
                q.pop();//出队
                ar.push_back(p->val);//让数据进入
                if(p->left)//不为空入队
                    q.push(p->left);
                if(p->right)
                    q.push(p->right);
            }
            arr.push_back(ar);//将这一层的数据放入arr中
            qsize = q.size();//下一层的结点个数
        }
        reverse(arr.begin(),arr.end());
        return arr;
    }
};

相关文章
|
1月前
【LeetCode 31】104.二叉树的最大深度
【LeetCode 31】104.二叉树的最大深度
19 2
|
1月前
【LeetCode 29】226.反转二叉树
【LeetCode 29】226.反转二叉树
15 2
|
1月前
【LeetCode 28】102.二叉树的层序遍历
【LeetCode 28】102.二叉树的层序遍历
15 2
|
1月前
【LeetCode 43】236.二叉树的最近公共祖先
【LeetCode 43】236.二叉树的最近公共祖先
18 0
|
1月前
【LeetCode 38】617.合并二叉树
【LeetCode 38】617.合并二叉树
14 0
|
1月前
【LeetCode 37】106.从中序与后序遍历构造二叉树
【LeetCode 37】106.从中序与后序遍历构造二叉树
16 0
|
1月前
【LeetCode 34】257.二叉树的所有路径
【LeetCode 34】257.二叉树的所有路径
12 0
|
1月前
【LeetCode 32】111.二叉树的最小深度
【LeetCode 32】111.二叉树的最小深度
16 0
|
3月前
|
存储 算法
二叉树进阶-学会层序遍历助你一次刷完leetcode10道题
文章深入探讨了二叉树的层序遍历方法,并展示了如何通过队列实现层序遍历的算法逻辑,同时指出掌握层序遍历技巧可以帮助解决LeetCode上的多道相关题目。
二叉树进阶-学会层序遍历助你一次刷完leetcode10道题
|
3月前
|
算法 Java
LeetCode第94题二叉树的中序遍历
文章介绍了LeetCode第94题"二叉树的中序遍历"的解法,使用递归实现了中序遍历的过程,遵循了"左根右"的遍历顺序,并提供了清晰的Java代码实现。
LeetCode第94题二叉树的中序遍历