题目
给定一个 N 叉树,返回其节点值的层序遍历。 (即从左到右,逐层遍历)。
例如,给定一个 3叉树 :
返回其层序遍历:
[
[1],
[3,2,4],
[5,6]
]
说明:
树的深度不会超过 1000。
树的节点总数不会超过 5000。
思路
1、广度优先搜索。
实现方法
一、广度优先搜索
/*
// Definition for a Node.
class Node {
public:
int val;
vector<Node*> children;
Node() {}
Node(int _val, vector<Node*> _children) {
val = _val;
children = _children;
}
};
*/
class Solution {
public:
vector<vector<int>> levelOrder(Node* root) {
vector<vector<int>> res;
if(!root) return res;
queue<Node*> q;
q.push(root);
while(!q.empty()){
vector<int> r;
int count=q.size();
while(count>0){
Node* top=q.front();
r.push_back(top->val);
q.pop();
count--;
if(top->children.size()!=0){
for(auto k:top->children)
q.push(k);
}
}
res.push_back(r);
}
return res;
}
};