题目描述
给定一个 N 叉树,返回其节点值的后序遍历。
- 示例:给定一个 3叉树 :
返回其后序遍历: [5,6,3,2,4,1].
说明: 递归法很简单,你可以使用迭代法完成此题吗?
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/n-ary-tree-postorder-traversal
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题思路
思路一:递归。思路比较简单,可直接看代码。
思路二:迭代。再利用一下示例。
其后序遍历为[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) {
val = _val;
}
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;
}
// 遇到叶子节点就添加至list中,注意一定要return
if(root.children.isEmpty()) {
list.add(root.val);
return list;
}
// 对孩子节点进行递归
for(Node node : root.children) {
postorder(node);
}
// 孩子节点递归完最后添加当前父节点
list.add(root.val);
return list;
}
}
思路二:迭代
class Solution {
public List<Integer> postorder(Node root) {
List<Integer> list = new ArrayList<>(); // 输出结果
Stack<Node> stack = new Stack<>(); // 栈
if(root == null) {
return list;
}
stack.push(root); // 先把根节点压栈
while(!stack.isEmpty()) {
Node node = stack.pop();
list.add(0, node.val); // 每次将栈顶弹栈,并将其val放在list的首位
for(Node n : node.children) { // 孩子节点按照从左往右的顺序压栈,如此一来弹栈顺序便是从右往左
stack.push(n);
}
}
return list;
}
}
注意点
- 注意是每次均放在list的首位。
- 友情链接:第589题——N叉树的前序遍历。