N叉树的后序遍历

下面这个是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,
在这里插入图片描述
反转后就是后序遍历的顺序

发布了213 篇原创文章 · 获赞 22 · 访问量 6万+

猜你喜欢

转载自blog.csdn.net/weixin_43113679/article/details/100714108