题解 | #二叉树的后序遍历#
二叉树的后序遍历
http://www.nowcoder.com/practice/1291064f4d5d4bdeaefbf0dd47d78541
迭代思路细节挺重要
import java.util.*;
/*
* public class TreeNode {
* int val = 0;
* TreeNode left = null;
* TreeNode right = null;
* public TreeNode(int val) {
* this.val = val;
* }
* }
*/
public class Solution {
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
*
* @param root TreeNode类
* @return int整型一维数组
*/
ArrayList<Integer> res = new ArrayList<>();
public int[] postorderTraversal (TreeNode root) {
// write code here
if(root==null) return new int[0];
// dfs(root);
Stack<TreeNode> sk = new Stack<>();
//定义两个指针,t指向栈顶,c指向刚出栈的节点,初始都指向root
TreeNode t = root;
TreeNode c = root;
sk.push(t);
while(!sk.isEmpty()){
//t赋值为栈顶元素
t = sk.peek();
//判断左子树是否为空,以及上一个弹出节点是否为左子树,以及是否为右子树
if(t.left!=null&&c!=t.left&&c!=t.right){
//将左子树入栈
sk.push(t.left);
//判断右子树是否为空或者上一个节点是否为右子树
}else if(t.right!=null&&c!=t.right){
sk.push(t.right);
}else{
res.add(t.val);
sk.pop();
c = t;
}
}
int len = res.size();
int[] result = new int[len];
for(int i = 0;i < len;i++){
result[i] = res.get(i);
}
return result;
}
}