给定一个 n 叉树的根节点 root ,返回 其节点值的 后序遍历 。
n 叉树 在输入中按层序遍历进行序列化表示,每组子节点由空值 null 分隔(请参见示例)。
示例 1:
输入:root = [1,null,3,2,4,null,5,6]
输出:[5,6,3,2,4,1]
示例 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]
输出:[2,6,14,11,7,3,12,8,4,13,9,10,5,1]
提示:
节点总数在范围 [0, 104] 内
0 <= Node.val <= 104
n 叉树的高度小于或等于 1000
进阶:递归法很简单,你可以使用迭代法完成此题吗?
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/n-ary-tree-postorder-traversal
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
* 解题思路: * 参考589,只修改list.add()的位置即可
public class Solution590 {
public List postorder(Node root) {
List list = new ArrayList<>();
if (root == null) {
return list;
}
preorder2(root, list);
return list;
}
//递归法
public void preorder2(Node root, List list) {
for (Node node : root.children) {
preorder2(node, list);
}
list.add(root.val);
}
//迭代法
public void preorder3(Node root, List list) {
Map levelList = new HashMap<>();
Map levelIndexList = new HashMap<>();
int level = 0;
levelList.put(0, root);
levelIndexList.put(0, 0);
while (true) {
Node node = levelList.get(level);//遍历的永远是最后一级节点
List children = node.children;
int index = levelIndexList.get(level);
if (children.size() == 0 || index == children.size()) {
levelList.remove(level);
level--;
//加入节点值
//遍历层级为负数则代表结束了
if (level < 0) {
break;
}
Node node1 = levelList.get(level);
list.add(node1.val);
levelIndexList.put(level, levelIndexList.get(level) + 1);//上一级的index+1
} else {
Node node1 = children.get(index);
if (node1 == null) {
levelIndexList.put(level, index + 1);
continue;
}
level++;
levelList.put(level, node1);
levelIndexList.put(level, 0);
}
}
list.add(root.val);
}
}