LeetCode102. 二叉树的层序遍历

最新推荐文章于 2024-07-02 20:59:51 发布

快乐早睡

最新推荐文章于 2024-07-02 20:59:51 发布

阅读量570

点赞数 25

分类专栏： LeetCode 文章标签：算法 leetcode 数据结构 c++ 二叉树层序遍历

本文链接：https://blog.csdn.net/weixin_40909461/article/details/138368265

版权

LeetCode 专栏收录该内容

132 篇文章 0 订阅

订阅专栏

在这里插入图片描述

LeetCode102. 二叉树的层序遍历

🍕🎉🥓🎊🚗🐷😱❤😀❣🥙🎇🕵️‍♀️😎🎑🎁🎉😚🎇

题目：

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

示例：

在这里插入图片描述

输入： $roo t = [3, 9, 20, n u ll, n u ll, 15, 7]$
输出： $[[3], [9, 20], [15, 7]]$

解题思路：

利用队列，每遍历一个元素，就将其加入到临时数组res_temp中，然后查看其是否有左右孩子，若有的话则入队。

C++

class Solution {
public:
    vector<vector<int>> levelOrder(TreeNode* root) {
        queue<TreeNode*> q;
        vector<vector<int>> res;
        if(root != NULL) q.push(root);
        while(!q.empty()){
            vector<int> res_temp;
            int size = q.size();
            for(int i = 0; i < size; i++){
                TreeNode* temp = q.front();
                q.pop();
                res_temp.push_back(temp->val);
                if(temp->left) q.push(temp->left);
                if(temp->right) q.push(temp->right);
            }
            res.push_back(res_temp);
        }
        return res;
    }
};

Java版本

class Solution {
    public List<List<Integer>> levelOrderBottom(TreeNode root) {
        List<List<Integer>> resList = new ArrayList<List<Integer>>();
        if(root == null) return resList;
        Queue<TreeNode> q = new LinkedList<TreeNode>();
        q.offer(root);
        while(!q.isEmpty()){
            List<Integer> temp = new ArrayList<Integer>();
            int len = q.size();
            while(len > 0){
                TreeNode tmpNode = q.poll();
                temp.add(tmpNode.val);
                if(tmpNode.left != null) q.offer(tmpNode.left);
                if(tmpNode.right != null) q.offer(tmpNode.right);
                len--;
            }
            resList.add(temp);
        }
        return resList;
    }
}