思路:
本质相同,也是考察二叉树层序遍历算法,这里用DFS广度优先实现。只不过每个结点有N个孩子(或空或非空)。从本题学习到了类中二维容器成员用法,Node类中成员children是一个一维容器,存放的都是Node*结点,遍历时只需利用容器size函数遍历即可。
具体代码注释如下:
/*
// 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 {
public:
vector<vector<int>> levelOrder(Node* root) {
vector<vector<int>>ans;//答案二维容器
queue<Node*>q;
if(root){q.push(root);}//根结点非空则入队
while(!q.empty()){//队列非空时
int size=q.size();
vector<int>ret;
for(int i=0;i<size;i++){//遍历本层
Node*cur=q.front();
q.pop();
if(cur){ret.push_back(cur->val);}//放入一维容器
for(int i=0;i<cur->children.size();i++){//当前结点的N个孩子非空则入队
q.push(cur->children[i]);
}
}
ans.push_back(ret);//一层遍历结束放入二维容器
}
return ans;
}
};