LeetCode 之 Binary Tree Level Order Traversal I and II-CSDN博客

本文链接：https://blog.csdn.net/lihdchao/article/details/50038495

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就行了。还有其他解法，有空再想吧。