590. N 叉树的后序遍历
给定一个 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
进阶:递归法很简单,你可以使用迭代法完成此题吗?
分析
今天的每日一题。
迭代法的思路可以参考代码随想录的 二叉树的统一迭代法:
https://github.com/youngyangyang04/leetcode-master/blob/master/problems/%E4%BA%8C%E5%8F%89%E6%A0%91%E7%9A%84%E7%BB%9F%E4%B8%80%E8%BF%AD%E4%BB%A3%E6%B3%95.md
题解(Java)
递归:
/*
// 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 {
private List<Integer> ans = new ArrayList<>();
public List<Integer> postorder(Node root) {
dfs(root);
return ans;
}
private void dfs(Node root) {
if (root == null) {
return;
}
for (Node node : root.children) {
postorder(node);
}
ans.add(root.val);
}
}
迭代:
/*
// 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 {
public List<Integer> postorder(Node root) {
List<Integer> ans = new ArrayList<>();
if (root == null) return ans;
Deque<Node> stack = new LinkedList<>();
stack.push(root);
while (!stack.isEmpty()) {
Node node = stack.pop();
//节点按照中 右 左的顺序放入栈
if (node != null) {
stack.push(node);
//null是作为根节点和叶子节点的分隔线(所有的根节点跟叶子节点之间都有null)。
stack.push(null);
List<Node> children = node.children;
if (children != null) {
for (int i = children.size() - 1; i >= 0; i--) {
stack.push(children.get(i));
}
}
} else {
//所有的根节点跟叶子节点之间都有null,pop出来的即为根节点,包括树的最底层。
node = stack.pop();
ans.add(node.val);
}
}
return ans;
}
}