给定一个 N 叉树,返回其节点值的 前序遍历 。
N 叉树 在输入中按层序遍历进行序列化表示,每组子节点由空值 null 分隔(请参见示例)。
进阶:
递归法很简单,你可以使用迭代法完成此题吗?
示例 1:
输入:root = [1,null,3,2,4,null,5,6]
输出:[1,3,5,6,2,4]
示例 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,6,7,11,14,4,8,12,5,9,13,10]
提示:
- N 叉树的高度小于或等于 1000
- 节点总数在范围 [0, 10^4] 内
递归
class Solution {
private List<Integer> res = new ArrayList<>();
public List<Integer> preorder(Node root) {
doWork(root);
return res;
}
private void doWork(Node root) {
if (null == root) {
return;
}
res.add(root.val);
if (null != root.children && !root.children.isEmpty()) {
for (Node child : root.children) {
doWork(child);
}
}
}
}
迭代
class Solution2 {
public List<Integer> preorder(Node root) {
List<Integer> res = new ArrayList<>();
Stack<Node> stack = new Stack<>();
stack.push(root);
while (!stack.isEmpty()) {
Node t = stack.pop();
res.add(t.val);
if (null != t.children && !t.children.isEmpty()) {
for (int i = t.children.size() - 1 ; i >= 0 ; i--) {
stack.push(t.children.get(i));
}
}
}
return res;
}
}