您的位置:首页 > 其它

[LeetCode] Binary Tree Level Order Traversal II

2017-09-10 20:40 267 查看
[Problem]
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],
]

[Solution]
class Solution {
public:
vector<vector<int> > levelOrderBottom(TreeNode *root) {
// Start typing your C/C++ solution below
// DO NOT write int main() function

// result
vector<vector<int> > res;
if(NULL == root)return res;

// store tmp result in stack
stack<vector<int> > mStack;
queue<vector<TreeNode*> > mQueue;

// add root node
vector<TreeNode*> tmp;
tmp.push_back(root);
mQueue.push(tmp);

// level order
while(!mQueue.empty()){
vector<int> levelRes;
vector<TreeNode*> levelNode;

// iterate
vector<TreeNode*> nodes = mQueue.front();
vector<TreeNode*>::iterator it;
for(it = nodes.begin(); it != nodes.end(); ++it){
levelRes.push_back((*it)->val);
if((*it)->left != NULL){
levelNode.push_back((*it)->left);
}
if((*it)->right != NULL){
levelNode.push_back((*it)->right);
}
}
mStack.push(levelRes);
mQueue.pop();
if(levelNode.size() > 0){
mQueue.push(levelNode);
}
}

// result
while(!mStack.empty()){
res.push_back(mStack.top());
mStack.pop();
}
return res;
}
};

说明:版权所有,转载请注明出处。Coder007的博客
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: