LeetCode 之 Binary Tree Level Order Traversal I and II

最新推荐文章于 2025-09-18 03:57:00 发布

原创最新推荐文章于 2025-09-18 03:57:00 发布 · 265 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#遍历 #leetcode

leetcode 专栏收录该内容

44 篇文章

订阅专栏

本文介绍了一种解决二叉树底部向上层次遍历的方法，通过队列迭代遍历二叉树节点，并按层次顺序收集节点值。文章提供了一个具体的C++实现案例，对比了不同实现方法的效率。

Given a binary tree, return the bottom-up level order traversal of its nodes' values. (ie, from left to right, level by level from leaf to root).

For example:
Given binary tree {3,9,20,#,#,15,7},

return its bottom-up level order traversal as:

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

这个是第II题，主要考察的二叉树的遍历，还是采用队列，迭代遍历方式。不过这次我们要能判断不同的层次，所以在不同层间加入一个NULL元素作为标志，代码如下：

class Solution {
public:
    vector<vector<int>> levelOrderBottom(TreeNode* root) {
        vector<vector<int>> ans;
        if(root==NULL) return ans;
        queue<TreeNode*> tree;
        tree.push(root);
        tree.push(NULL);
        int flag=0;
        vector<int> level;
        while(!tree.empty()){
            auto temp=tree.front();
            if(temp!=NULL){
                if(flag){
                    ans.push_back(level);
                    level.clear();
                    level.push_back(temp->val);
                    flag=0;
                }
                else{
                    level.push_back(temp->val);
                }
                if(temp->left!=NULL) tree.push(temp->left);
                if(temp->right!=NULL) tree.push(temp->right);
                tree.pop();
                if(tree.empty()) return ans;
                if(tree.front()==NULL) tree.push(NULL);
            }
            else{
                tree.pop();
                flag=1;
            }
        }
        if(!level.empty()) 
            ans.push_back(level);
        reverse(ans.begin(),ans.end());
        return ans;
    }
};

需要注意的是返回的是从下到上的顺序，所以我们要把结果ans翻转一下。

在我最开始的想法中，是把每一层的vector插入到ans的第一个位置，这样得到的结果就是正确的。可是提交后发现很慢，差不多要60ms。我又换种方式，把level，push到ans的后面。遍历结束后，对ans做一次翻转，这样时间差不多只用到了8ms，和大部分人的时间差不多。
对于第I题，我们只要不在最后不翻转ans就行了。还有其他解法，有空再想吧。