LeetCode 之 Binary Tree Level Order Traversal I and II

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}

,

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