题目
给定一个二叉树,返回它的 后序 遍历。
进阶: 递归算法很简单,你可以通过迭代算法完成吗?
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/binary-tree-postorder-traversal
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解答
解法一:递归
递归解法不用多说,很简单。
代码
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public List<Integer> postorderTraversal(TreeNode root) {
List<Integer> res = new ArrayList<>();
postorderTraversal(root, res);
return res;
}
private void postorderTraversal(TreeNode node, List<Integer> res) {
if(node == null) return;
postorderTraversal(node.left, res);
postorderTraversal(node.right, res);
res.add(node.val);
}
}
结果
解法二:前序遍历逆序
我们观察到:
-
后序遍历的顺序是:左 -> 右 -> 根,倒着的顺序就是 根 -> 右 -> 左。
-
前序遍历的顺序是:根 -> 左 -> 右,因此我们只需要简单的更改左右子树的压栈顺序即可。
最后将遍历的结果逆序或者采用头插法,就是后序遍历的结果了。
代码
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public List<Integer> postorderTraversal(TreeNode root) {
if(root == null) return new ArrayList<>();
LinkedList<Integer> res = new LinkedList<>();
LinkedList<TreeNode> stack = new LinkedList<>();
stack.push(root);
while(!stack.isEmpty()) {
TreeNode node = stack.pop();
res.addFirst(node.val);
if(node.left != null) {
stack.push(node.left);
}
if(node.right != null) {
stack.push(node.right);
}
}
return res;
}
}
结果
解法三:完全非递归后序
pre 变量:记录前一个被访问的结点。
具体流程如下:
- 先把左结点全部压栈。
- 然后观察栈顶结点的右子树是否为 null 或者已经遍历过了。
- 如果是 null 或者右子树已经遍历过了,就把当前结点出栈。
- 否则当前栈顶结点不出栈,而是去遍历右子树。
代码
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public List<Integer> postorderTraversal(TreeNode root) {
if(root == null) return new ArrayList<>();
LinkedList<Integer> res = new LinkedList<>();
LinkedList<TreeNode> stack = new LinkedList<>();
TreeNode cur = root;
TreeNode pre = null;
while(cur != null || !stack.isEmpty()) {
while(cur != null) {
stack.push(cur);
cur = cur.left;
}
TreeNode node = stack.peek();
if(node.right == null || node.right == pre) {
stack.pop();
res.add(node.val);
pre = node;
cur = null;
} else {
cur = node.right;
}
}
return res;
}
}
结果
解法四:前中后序非递归通用解
对于二叉树中的任何一个节点而言,它都有两种角色:
- flag = true:作为值,此时可加入结果集。
- flag = false:作为根树,需要继续处理其左右子树。
对于前,中,后序遍历代码基本一致,仅有少量改动,类似于写递归代码的体验。
代码
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
class Entry {
TreeNode node;
boolean flag;
Entry(TreeNode node, boolean flag) {
this.node = node;
this.flag = flag;
}
}
public List<Integer> postorderTraversal(TreeNode root) {
if(root == null) return new ArrayList<>();
LinkedList<Integer> res = new LinkedList<>();
LinkedList<Entry> stack = new LinkedList<>();
stack.push(new Entry(root, false));
while(!stack.isEmpty()) {
Entry entry = stack.pop();
TreeNode node = entry.node;
boolean flag = entry.flag;
if(flag) {
res.add(node.val);
} else {
// 前中后序遍历只有这里稍微不同。
stack.push(new Entry(node, true));
if(node.right != null) stack.push(new Entry(node.right, false));
if(node.left != null) stack.push(new Entry(node.left, false));
}
}
return res;
}
}