难度评级:中等
给定一个 N 叉树,返回其节点值的层序遍历。(即从左到右,逐层遍历)。
树的序列化输入是用层序遍历,每组子节点都由 null 值分隔(参见示例)。
示例 1:
输入:root = [1,null,3,2,4,null,5,6] 输出:[[1],[3,2,4],[5,6]]
示例 2:
输入:root = [1,null,2,3,4,5,null,null,6,7,null,8,null,9,10,null,null,11,null,12,null,13,null,null,14] 输出:[[1],[2,3,4,5],[6,7,8,9,10],[11,12,13],[14]]
题解思路:DFS
本题由二叉树层序遍历推广而至,同样有DFS与队列两种思路,本文采用DFS,具体思路见二叉树题解,这里只需要将二叉树的左右孩子更改为全部孩子即可,通过for循环进行遍历。
题解代码如下:
/*
// 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>> res;
void pre_DFS(Node* root, int deep, vector<vector<int>>& res){
if (root == NULL){
return;
}
if (deep >= res.size()){
res.push_back(vector<int> {}); // 扩容
}
res[deep].push_back(root->val);
for (auto i : root->children){ // 遍历孩子
pre_DFS(i, deep + 1, res);
}
}
vector<vector<int>> levelOrder(Node* root) {
pre_DFS(root, 0, res);
return res;
}
};