题目:
给定一个 N 叉树,返回其节点值的层序遍历。 (即从左到右,逐层遍历)。
例如,给定一个 3叉树 :
返回其层序遍历:
[
[1],
[3,2,4],
[5,6]
]
说明:
树的深度不会超过 1000。
树的节点总数不会超过 5000。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/n-ary-tree-level-order-traversal
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
方法一:迭代法:
解题思路:
提到层序遍历,最普遍的方法就是树的广度优先遍历。这道题的难点在于如何按层输出,方法是在常规的广度优先遍历的while循环中再加入一层for循环,for循环的大小是当前的queue.size,在访问到树的第n层时就把第(n+1)层的子结点全部插入到队列queue中,则下次while循环是queue的大小就是第(n+1)层的所有结点,插入到一个临时vector< int > layer中即可。
时间复杂度: O(n) //每个结点遍历一次
空间复杂度: O(n) //queue
/*
// Definition for a Node.
class Node {
public:
int val;
vector<Node*> children;
Node() {}
Node(int _val) {
val = _val;
}
Node(int _val, vector<Node*> _children) {
val = _val;
children = _children;
}
};
*/
class Solution {
private:
vector<vector<int>> result;
public:
vector<vector<int>> levelOrder(Node* root) {
queue<Node*> q;
if(root != nullptr) q.push(root);
while(!q.empty()) {
int size = q.size();
vector<int> layer;
for(int i = 0; i < size; ++i) {
//第 n 层的结点全部插入vector
layer.push_back(q.front()->val);
for(Node* node_ : q.front()->children) //第 n+1 层的结点全部入队queue
q.push(node_);
q.pop();
}
result.push_back(layer);
}
return result;
}
};