题目:
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] ]
https://oj.leetcode.com/problems/binary-tree-level-order-traversal/
解题思路:这个没啥说的,用个队列直接就搞定了,前面已经有类似的了。
#include<iostream>
#include<vector>
#include<queue>
using namespace std;
struct TreeNode
{
int val;
TreeNode *left;
TreeNode *right;
TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};
vector<vector<int> > levelOrder(TreeNode *root)
{
vector<vector<int> >results;
if(root==NULL)
return results;
queue<TreeNode *> q;
q.push(root);
int k=1,level=1;//level表示当前层中的节点数
while(!q.empty())
{
vector<int> result;
k=level;
level=0;
for(int i=0;i<k;i++)
{
if(q.front()->left!=NULL)
{
q.push(q.front()->left);
level++;
}
if(q.front()->right!=NULL)
{
q.push(q.front()->right);
level++;
}
result.push_back(q.front()->val);
q.pop();
}
results.push_back(result);
}
return results;
}
int main()
{
TreeNode* root=new TreeNode(1);
root->left=new TreeNode(2);
root->right=new TreeNode(3);
root->left->left=new TreeNode(4);
root->right->left=new TreeNode(5);
vector<vector<int> > results=levelOrder(root);
system("pause");
return 0;
}