给定一个二叉树,返回它的 后序 遍历。
示例:
输入: [1,null,2,3]
1
\
2
/
3
输出: [3,2,1]
进阶: 递归算法很简单,你可以通过迭代算法完成吗?
遍历顺序为左右中
递归:从根节点开始,一直遍历它的左节点,直到节点为空时返回上一层,当前节点也就是中间节点,继续遍历该节点的右节点,当右节点遍历完成之后,返回上一层的中间节点然后添加当前节点值。
public static List<Integer> postorderTraversal(TreeNode root) {
List<Integer> res = new ArrayList<>();
inorder(root, res);
return res;
}
public static void inorder(TreeNode root, List<Integer> res) {
if (root == null) return;
if (root.left != null) inorder(root.left, res);
if (root.right != null) inorder(root.right, res);
res.add(root.val);
}
迭代:和前序遍历刚好相反,先进后出采用栈来存储节点,头节点是最先被遍历到但是是最后存储值的,可以直接放进栈中,循环弹出栈顶节点,存储节点值,把左节点和右节点压入栈中,下一次循环也就会先弹出右节点,把当前节点值存储到集合中的开始位置,然后再次进行遍历。
public static List<Integer> postorderTraversal(TreeNode root) {
LinkedList<Integer> output = new LinkedList<>();
if (root == null) return output;
Stack<TreeNode> stack = new Stack<>();
stack.push(root);
while (!stack.isEmpty()) {
TreeNode node = stack.pop();
output.addFirst(node.val);
if (node.left != null) {
stack.add(node.left);
}
if (node.right != null) {
stack.add(node.right);
}
}
return output;
}