leetcode第102题题解(二叉树的层序遍历)利用C++和Java解决

最新推荐文章于 2024-06-04 14:56:47 发布

原创最新推荐文章于 2024-06-04 14:56:47 发布 · 419 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#二叉树 #队列 #leetcode #java #算法

leetcode 专栏收录该内容

8 篇文章

订阅专栏

这篇博客介绍了如何使用C++和Java解决LeetCode第102题——二叉树的层序遍历。通过队列实现逐层遍历，从左到右访问每个节点。示例中给出了一棵二叉树，并展示了层次遍历的结果。解题策略是利用队列，当一个节点出队时，其左右子节点入队，直至队列为空。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述：
给你一个二叉树，请你返回其按层序遍历得到的节点值。（即逐层地，从左到右访问所有节点）。

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

在这里插入图片描述
返回其层次遍历结果：

[
[3],
[9,20],
[15,7]
]

解题思路：
我们做这道题是首先就可以想到队列来解决问题，开始根节点进入，随后，队列里有进有出，那么当一个数出去的时候，他的左节点和右节点依次进入，这样迭代下去
那么我们首先想到的就是二元组，但是为了简化，我们用一个变量node即可

C++代码

class Solution {
public:
    vector<vector<int>> levelOrder(TreeNode* root) {
        
        vector <vector <int>> ret;
        if (!root) return ret;

        queue <TreeNode*> q;
        q.push(root);
        while (!q.empty()) {
            int currentLevelSize = q.size();
            ret.push_back(vector <int> ());
            for (int i = 1; i <= currentLevelSize; ++i) {
                auto node = q.front(); q.pop();
                ret.back().push_back(node->val);
                if (node->left) q.push(node->left);
                if (node->right) q.push(node->right);
            }
        }
        
        return ret;
    }


    
};

Java代码

public List<List<Integer>> levelOrder(TreeNode root) {
        //按层遍历即可
        //1.
        List<List<Integer>> lists = new ArrayList<>();
        if (root == null) {
            return lists;
        }
        //2.
        List<TreeNode> nodes = new ArrayList<>();
        nodes.add(root);
        while (!nodes.isEmpty()) {
            int size = nodes.size();
            List<Integer> list = new ArrayList<>();
            for (int i = 0; i < size; i++) {
                TreeNode remove = nodes.remove(0);
                list.add(remove.val);
                if (remove.left != null) {
                    nodes.add(remove.left);
                }
                if (remove.right != null) {
                    nodes.add(remove.right);
                }
            }
            lists.add(list);
        }
        return lists;
    }