题目不用多说,就是二叉树的后序遍历。如果是递归进行遍历,相信大家是都会的。这里说下用栈进行遍历的大体思路及代码:
思路
与前序遍历不同,如果用栈实现后序遍历,当遍历完某个子树后,需要判断之前遍历的是左子树还是右子树。如果是左子树,则要再遍历右子树。如果刚遍历的是右子树,则要访问根节点,并出栈。
代码
public class Solution {
public List<Integer> postorderTraversal(TreeNode root) {
List<Integer> result = new ArrayList<Integer>();
if(root == null){
return result;
}
Deque<TreeNode> stack = new ArrayDeque<TreeNode>();
TreeNode prev = null;
while(true){
while(root != null){
stack.push(root);
root = root.left;
}
if(!stack.isEmpty()){
TreeNode top = stack.peek();
if(top.right == null || (prev != null && prev == top.right)){
prev = stack.pop();
result.add(prev.val);
}else{
root = top.right;
}
}else{
break;
}
}
return result;
}
}