原题描述:
Given a binary tree, return the level order traversal of its nodes' values. (ie, from left to right, level by level).
For example:
Given binary tree {3,9,20,#,#,15,7}
,
3
/ \
9 20
/ \
15 7
return its level order traversal as:
[
[3],
[9,20],
[15,7]
]
代码如下:
<span style="font-size:14px;">/**
* Definition for binary tree
* 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) {
vector<vector<int> > result;
if (!root) return result;
queue<TreeNode*> temp;
temp.push(root);
while (!temp.empty()) {
queue<TreeNode*> q;
vector<int> v;
while (!temp.empty()) {
TreeNode *t = temp.front();
v.push_back(t->val);
if(t->left) q.push(t->left);
if(t->right) q.push(t->right);
temp.pop();
}
result.push_back(v);
temp = q;
}
return result;
}
};</span>
NOTE:要判断root为空的情况,否则会出现runtime error!