题目描述
给定一个二叉树,返回它的 后序 遍历。
代码
迭代实现
/**
* 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> list = new ArrayList<>();
Stack<TreeNode> nodeStack = new Stack<>();
Stack<Integer> nodeState = new Stack<>(); // 记录右节点是否被访问过,1表示已经访问了,0表示没有访问
if (root == null) return list;
nodeStack.push(root);
nodeState.push(0);
root = root.left;
while(!nodeStack.isEmpty()){
while(root != null){
nodeStack.push(root);
nodeState.push(0);
root = root.left;
} // 当这个循环跳出的时候,说明nodeStack栈顶的那个节点没有左节点
if (nodeState.peek() == 1){ // 如果此时已经访问过右节点了,这个时候可以访问根节点
list.add(nodeStack.pop().val);
nodeState.pop(); // 把根节点对应的状态值去除
}else{
root = nodeStack.peek().right;
nodeState.pop(); // 更该状态值,由0变1
nodeState.push(1);
}
}
return list;
}
}