问题描述
给定一个N叉树,返回其节点值的后序遍历。
例如,给定一个 3叉树 :
返回其后序遍历: [5,6,3,2,4,1].
说明: 递归法很简单,你可以使用迭代法完成此题吗?
思路
与二叉树的后序遍历的思路一致,
前序、后序,中序,只不过打印数字或者存数字的位置不同而已
实现
/*
// Definition for a Node.
class Node {
public int val;
public List<Node> children;
public Node() {}
public Node(int _val,List<Node> _children) {
val = _val;
children = _children;
}
};
*/
class Solution {
public List<Integer> postorder(Node root) {
List<Integer> list = new ArrayList<>();
if (root == null) {
return list;
}
for (Node node : root.children) {
list.addAll(postorder(node));//将字节的所有孩子节点都存入其中
}
list.add(root.val);//把根节点也存进去
return list;
}
}
优化
/*
// Definition for a Node.
class Node {
public int val;
public List<Node> children;
public Node() {}
public Node(int _val,List<Node> _children) {
val = _val;
children = _children;
}
};
*/
class Solution {
List<Integer> list = new ArrayList<>();
public List<Integer> postorder(Node root) {
if (root == null) {
return list;
}
for (Node node : root.children) {
postorder(node);//起到遍历到最后一个元素的作用,存入在最后做
}
list.add(root.val);//遍历到最后一个时候,就会存了,不需要每次都存节点所有子节点,如上种做法那样
return list;
}
}
小结
与前序遇到的问题是一致的,注意static的含义,用法。