问题不描述了,后续遍历二叉树,递归方法同leetcode7.(改下顺序)
非递归方法
package leetcode6and7.BinaryTree;
import java.util.ArrayList;
import java.util.Stack;
public class Solution {
public ArrayList<Integer> postorderTraversal(TreeNode root){
ArrayList<Integer> arrayList = new ArrayList<Integer>();
if(root == null)
return arrayList;
Stack<TreeNode>stack = new Stack<TreeNode>();
stack.push(root);
TreeNode prenote=null;
while(!stack.empty()){
TreeNode current = stack.peek();//look at the top of the stack,not rm it!
//以下为仅有根节点,或者父节点第一次入栈
if(prenote==null||prenote.left==current||prenote.right==current){//prenote==null then just root;
if(current.left !=null){
stack.push(current.left);
}else if(current.right!=null){
stack.push(current.right);
}else{
stack.pop();
arrayList.add(current.val);
}
}else if(current.left==prenote){//此时第二次遍历到父节点,判断右子树是否有值,有值入栈,无值返回父节点
if(current.right!=null){
stack.push(current.right);
}else{
stack.pop();
arrayList.add(current.val);
}
}else if(current.right == prenote){//右子树已处理,处理父节点
stack.pop();
arrayList.add(current.val);
}
prenote=current;
}//while
return arrayList;
}
}
总结:上学那会就很头疼这个,还是要回头多想多思考该方法
总体思路:首先考虑特殊情况,即只有一个根节点的情况,即prenote==null时的情况,接下去考虑正常情况,先向左子树遍历,遍历到最左下的节点,判断其是否有右子树,如有则入栈继续判断,如无则退栈保存当前最左下节点,处理完该节点,返回该节点(记为A)的父节点,此时A的父节点第二次被处理,判断其是否有右子树,即重复上述过程。
注意的是父节点会被俩次处理