Leetcode 102. Binary Tree Level Order Traversal

最新推荐文章于 2023-11-06 11:13:49 发布

原创最新推荐文章于 2023-11-06 11:13:49 发布 · 171 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 专栏收录该内容

19 篇文章

订阅专栏

博客围绕二叉树层次遍历问题展开，给出题目链接及示例。解题方案有迭代和递归两种。迭代法用队列遍历，设置last和nlast变量完成换行；递归法也给出了相应执行用时和内存消耗情况。

题目描述

题目链接：https://leetcode-cn.com/problems/binary-tree-level-order-traversal/
给定一个二叉树，返回其按层次遍历的节点值。（即逐层地，从左到右访问所有节点）。
例如:
给定二叉树: [3,9,20,null,null,15,7],

返回其层次遍历结果：

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

解题方案

迭代
设置一个队列，对二叉树进行层次遍历，遍历到的节点压入队列；弹出某个节点后将其左右孩子节点依次压入队列中。
设置两个变量，记为last，nlast，其中last用于记录当前行最右节点，nlast记录下一行最右节点（nlast始终记录最新进入队列的节点）。当last等于最新打印的节点，则是last=nlast，这样就完成了换行操作。

class Solution {
public:
    vector< vector<int> > levelOrder(TreeNode* root) {
        vector<vector<int>> res;
        if(!root)
            return res;
        vector<TreeNode *> level;
        level.push_back(root);
        while(!level.empty())
        {
            vector<TreeNode *> temp;
            vector<int> level_val;
            for(int i=0;i<level.size();++i)
            {
                if(level[i]->left)
                    temp.push_back(level[i]->left);
                if(level[i]->right)
                    temp.push_back(level[i]->right);
                level_val.push_back(level[i]->val);
            }
            res.push_back(level_val);
            level = temp;
        
        }
        return res;
    }
};

执行用时16ms，内存消耗14MB。

递归

class Solution {
public:
    vector<vector<int>> levelOrder(TreeNode* root) {
        vector<vector<int>> res;
        preorder(root, 0, res);
        return res;
    }
    void preorder(TreeNode *root, int depth, vector<vector<int>> &res)
    {
        if(!root)
            return;
        if(depth>=res.size())
            res.push_back(vector<int> {});
        res[depth].push_back(root->val);
        preorder(root->left, depth+1, res);
        preorder(root->right, depth+1, res);
            }
        };

执行用时16ms，内存消耗14.7MB。