LeetCode 102. Binary Tree Level Order Traversal 解题报告

最新推荐文章于 2024-02-12 01:19:39 发布

原创最新推荐文章于 2024-02-12 01:19:39 发布 · 506 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode

编程解题专栏收录该内容

88 篇文章

订阅专栏

本文介绍了解决LeetCode102题的方法，该题要求实现二叉树的层序遍历并返回每层节点的值。通过广度优先搜索(BFS)实现，使用队列来跟踪节点并记录其深度，以便按层级顺序输出节点值。

LeetCode 102. Binary Tree Level Order Traversal 解题报告

题目描述

Given a binary tree, return the level order traversal of its nodes’ values. (ie, from left to right, level by level).

示例

Example 1:
Given binary tree [3,9,20,null,null,15,7]
input
return its level order traversal as:
output

注意事项

没有明确给出。

解题思路

我的思路：

这道题需要逐层把二叉树的节点值保存起来。其实就是二叉树的层序遍历，使用的是广度优先搜索（Breadth First Search, BFS）。由于需要逐层保存节点的值，所以遍历时额外维护一个深度信息，表示目前访问到的层数，一旦出现目前访问层数小于节点层数，就表示开始下一层的遍历，此时需要把上一层的节点信息保存到结果中。实现方式是普通的BFS，使用队列保存待访问到的节点，每次取出队列头部的节点进行访问，并在有子节点的情况下，将子节点入队，在队列为空时结束遍历。见下面的实现代码。
另一道题LeetCode 107. Binary Tree Level Order Traversal II是要以相反的顺序保存各层的节点值，即先保存最底层的节点值，最后保存根节点的值，只要在这道题的代码中加入一句reverse(result.begin(), result.end())就能通过。

代码

我的代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    vector<vector<int>> levelOrder(TreeNode* root) {
        int curDepth = 0;
        vector<vector<int>> result;
        vector<int> temp;
        queue<pair<TreeNode*, int>> q;

        if (root) {
            q.push({root, 0});

            while(q.size()) {
                pair<TreeNode*, int> cur = q.front();
                q.pop();

                if (curDepth < cur.second) {
                    result.push_back(temp);
                    temp.clear();
                    curDepth = cur.second;
                }

                temp.push_back((cur.first)->val);

                if ((cur.first)->left)
                    q.push({(cur.first)->left, curDepth + 1});

                if ((cur.first)->right)
                    q.push({(cur.first)->right, curDepth + 1});
            }

            result.push_back(temp);
        }

        return result;
    }
};