剑指 Offer . 从上到下打印二叉树 II

最新推荐文章于 2025-12-02 15:32:05 发布

原创最新推荐文章于 2025-12-02 15:32:05 发布 · 40 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#算法 #c++ #数据结构

该代码实现了一个二叉树的层次遍历算法，使用队列存储节点并逐层处理。遍历过程中，每层的节点值被添加到一个临时数组中，然后这个数组被添加到结果数组，最后返回结果数组，展示每一层的节点值。

从上到下按层打印二叉树，同一层的节点按从左到右的顺序打印，每一层打印到一行。

例如: 给定二叉树: [3,9,20,null,null,15,7],

    3
   / \
  9  20
    /  \
   15   7
返回其层次遍历结果：

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

代码如下：

class Solution {
public:
    vector<vector<int>> levelOrder(TreeNode* root) {
        vector<vector<int>> res;//定义结果数组
        queue<TreeNode*> q;//定义队列
        if(root==NULL)
        {
            return res;//当二叉树中没有节点之后，返回结果数组
        }
        q.push(root);
        while(!q.empty())
        {
            vector<int> temp;//建立一个临时数组，用于存储当前层打印结果
            for(int i=q.size();i>0;i--)//循环次数为当前层节点数，即队列的长度->每遍历完一层，将下一层的节点存入队列中
            {
                TreeNode* node=q.front();//队首元素出队，记为node
                q.pop();
                temp.push_back(node->val);//将node的值存入temp中
                if(node->left!=NULL)//node的左节点不为空，将左节点存入队列中
                {
                    q.push(node->left);
                }
                if(node->right!=NULL)//node的右节点不为空，将右节点存入队列中
                {
                    q.push(node->right);
                }
            }
            res.push_back(temp);//将临时数组存入结果数组中
        }
        return res;//返回结果数组

    }
};