1. 问题描述
给定一个 n 叉树的根节点 root ,返回 其节点值的 后序遍历 。
n 叉树 在输入中按层序遍历进行序列化表示,每组子节点由空值 null 分隔。
2. 解题思路
以输入root = [1,null,3,2,4,null,5,6],输出 [5,6,3,2,4,1] 为例:
1
/ | \
3 2 4
/ \
5 6
① 根节点 1 入队 list
② 当队列 list 不为空时,进入 while 循环:首先取出队尾元素也就是节点 1 并暂存至临时节点 temp 中,for 遍历该节点的子节点(节点 3、节点 2、节点 4),并将这些节点依次加入到 list 队尾,此时队中元素从队首到队尾
依次为节点 3、节点 2、节点 4。结果集 res 将临时节点存储的值加入到结果集的队首即此时结果集 res 队首元素为 1。
此时 res = [1]
③ 由于此时队伍 list 不为空(存在节点 3、节点 2、节点 4),所以继续进入 while 循环,首先取出队尾的节点 4 并暂存至临时节点 temp 中,for循环遍历其子节点,并将这些节点依次加入 list 队尾(实际节点 4 并无子节点)。结果集 res 将临时节点存储的值加入到结果集的队首即此时结果集 res 队首元素为 4。
此时 res = [4, 1]
④ 节点 2 和节点 4 同理,因为二者都是没有子节点的。
此时 res = [2, 4, 1]
⑤ 由于此时队伍 list 还剩下节点 3 即不为空,所以继续进入 while 循环,首先取出队尾的节点 3 并暂存至临时节点 temp 中,for循环遍历其子节点(节点 5、节点 6),并将这些节点依次加入 list 队尾。结果集 res 将临时节点存储的值加入到结果集的队首即此时结果集 res 队首元素为 3。
此时 res = [3, 2, 4, 1]
⑥ 由于 list 中还有节点 5、节点 6,重复节点 2(节点 4)的操作,依次将节点 6、节点 5加入结果集队首。
此时 res = [6, 3, 2, 4, 1]
此时 res = [5, 6, 3, 2, 4, 1]
3. 代码实现
class Solution {
// 存储结果集
LinkedList<Integer> res = new LinkedList<>();
public List<Integer> postorder(Node root) {
if (root == null) {
return res;
}
LinkedList<Node> list = new LinkedList<>();
list.add(root);
while (!list.isEmpty()) {
// 队尾元素取出并暂存在节点temp
Node temp = list.pollLast();
for (Node node: temp.children) {
list.addLast(node);
}
// 节点值添加到队首
res.addFirst(temp.val);
}
return res;
}
}