后续遍历--递归与非递归(java版)

版权声明::本文为博主原创文章,欢迎转载,博客地址: https://blog.csdn.net/u012535132/article/details/84330387

先访问左右孩子,再访问根节点。同样还是采用栈的形式,但是问题是,先访问左孩子出栈,根节点不能删除,再访问右孩子出栈,最后访问根节点出栈。我们的思路是这样的:

对于某个节点p:

   1)将p压入栈中,并将p所有的左孩子,全部压入栈中:

    while(stk.isEmpty() == false) {            
            if(p!=null && p.left!=null) {
                stk.add(p.left);
                p = p.left;
            }else
               .....
    }

      2)将栈顶元素弹出,访问。如果没有右孩子,栈顶元素继续出栈。如果有右孩子,将右孩子压入栈中,pVisit标记其右孩子,表示已经访问过了。当右孩子访问过之后,回到 1 节点(如下图),p.right = pVisit,表示右节点已经访问过了。

4)直到stack.isEmpty == true,则遍历结束;

需要一个辅助节点pRight,它专门指向已经访问过的右节点,例如(结合程序看):

import java.util.Stack;

class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;
    public TreeNode(int val) {
        this.val = val;
    }
}
public class Main {
	//递归求解
	public static void postOrder(TreeNode root) {
		if(root == null)return;
		postOrder(root.left);
		postOrder(root.right);
		System.out.print(root.val +" ");
	}
	//非递归求解
	public static void postOrder2(TreeNode root) {
		if(root == null)return;
		TreeNode p = root;
		TreeNode pVisit = null;
		Stack<TreeNode> stk = new Stack<TreeNode>();
		stk.add(p);
		
		while(stk.isEmpty() == false) {
			//只要你有左孩子,就将左孩子压入栈中
			if(p!=null && p.left!=null) {
				stk.add(p.left);
				p = p.left;
			}else {
				p = stk.peek();//栈顶元素,先出栈,可能还有右孩子
				if(p.right==null  || p.right==pVisit) {//如果没有右孩子或右孩子已经访问过了,出栈
					System.out.print(p.val+" ");
					pVisit = p;//这个很重要,考虑一下只有右孩子的树,得不断的回溯
					p = null;//没有新节点加入,继续进行出栈操作
					stk.pop();
				}else {//如果有右孩子,右孩子入栈
					pVisit = p.right;
					stk.add(p.right);
					p = p.right;
				}
			}
		}
	}
	public static void main(String[] args) {
		// TODO Auto-generated method stub
		TreeNode root = new TreeNode(1);
		 root.left = new TreeNode(2);
		 root.right = new TreeNode(3);
		 
		 root.left.left = new TreeNode(4);
		 root.left.right = new TreeNode(5);
		 
		 root.right.left = new TreeNode(6);
		 root.right.right = new TreeNode(7);
		
		 postOrder2(root);
	}

}

猜你喜欢

转载自blog.csdn.net/u012535132/article/details/84330387