题目: 给定一个 N 叉树,返回其节点值的后序遍历。
示例:
思路–迭代:
1.迭代方法中必定会涉及一个大循环,同时有必要借助栈,简便方法即借助一个栈即可完成目标任务。
2.分析 栈 在此题中的用法:以上图为例,1先进栈,然后出栈,遍历该结点的子结点为3、2、4,然后按照栈的特点,先进后出,则4先出,然后遍历其宝宝结点,若宝宝结点还有宝宝结点,则继续向下一层遍历(原理同于由1向3、2、4迭代)。直接将值加入res正好是相反的顺序,故需要进行翻转,或是在把stack中pop出来的值直接addFirst加入res之中。
Code-迭代
class Solution {
public List<Integer> postOrder(Node root) {
List<Integer> res = new LinkedList<>();
if (root == null) {
return res;
}
Stack<Node> stack = new Stack<>();
stack.push(root);
while (!stack.isEmpty()) {
Node cur = stack.pop(); //出栈的是栈顶元素,故是从右向左进行遍历
res.add(cur.val);
for (var child : cur.children) {
stack.push(child);
}
}
Collections.reverse(res); //进组元素刚好是倒序,进行翻转即可
return res;
}
}
思路-递归
1.递归算法中大多必须设计一个辅助函数,在辅助函数中反复调用辅助函数。(此题简单,不过多分析,直接上代码)
Code-递归
class Solution {
List<Integer> res = new LinkedList<>();
public List<Integer> postOrder(Node root) {
help(root);
return res;
}
public void help(Node root) { //辅助函数help
if (root == null) { //递归结束条件
return;
}
for (var child : root.children) {
help(child); //反复调用辅助函数
}
res.add(root.val); 将当前root.val加入res
}
}