下面这个是leetcode中第590题
给定一个 N 叉树,返回其节点值的后序遍历。
例如,给定一个 3叉树 :
返回其后序遍历: [5,6,3,2,4,1].
说明: 递归法很简单,你可以使用迭代法完成此题吗?
下面就是一位大神的答案,好理解
class Node {
public int val;
public List<Node> children;
public Node() {}
public Node(int _val,List<Node> _children) {
val = _val;
children = _children;
}
};
public class Main {
public List<Integer> postorder(Node root) {
List<Integer> res_pre = new ArrayList<>();
if(root == null)
return res_pre;
Stack<Node> s = new Stack<>();
s.push(root);
while(!s.isEmpty()){ //当栈中有元素说明还没有遍历完
Node n = s.pop();
res_pre.add(n.val); //往集合里放当前节点的值
for(Node node : n.children) //直接把子节点的集合遍历加入栈中
s.push(node);
}
Collections.reverse(res_pre); //集合中的值反转,
return res_pre;
}}
解释一下上面的过程,就以上面的图为例子,
首先把根节点放入集合,并且放进栈中,再弹出栈,遍历它的子节点集合,压入栈,这样压入栈的顺序就是3,2,4。
栈下一个弹出的就是4,把4节点的值放入集合,再遍历它的子节点集合,因为没有,直接下一个,
再把2的值放入集合,再遍历子节点集合放入栈,因为没有所以就下一个,
到3了,和上面一样,把5,6压入栈,
重复上面的步骤,这时候集合中的值的顺序是1,4,2,3,6,5,
反转后就是后序遍历的顺序