中序遍历--递归和非递归（java版）

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

根据中序遍历的顺序，对于任一结点，优先访问其左孩子，而左孩子结点又可以看做一根结点，然后继续访问其左孩子结点，直到遇到左孩子结点为空的结点才进行访问，然后按相同的规则访问其右子树。因此其处理过程如下：

　　对于任一结点root，引入一个辅助节点p，其作用是：标记已经访问过的节点，

  　1)将root压入栈中，只有有左孩子，就压入栈中

          if(p!=null &&  p.left!=null) {
                stk.add(p.left);
                p = p.left;
            }

 　 2)对栈顶元素 q 进行出栈操作，访问该栈顶结点。

        如果 q 没有右孩子，将辅助节点 p  设置为null。表示下一步还是进行2）操作，出栈操作

        如果 q 有右孩子，将右孩子压栈中，p = q的右孩子

                p = stk.pop();//弹出栈顶节点  左孩子--->根节点
                System.out.print(p.val+" ");//访问
                if(p!=null && p.right!=null) {//如果栈点元素有右孩子的话，将有节点压入栈中
                    stk.add(p.right);
                    p = p.right;
                }else
                    p = null;//p=stk.pop;已经访问过p了，p设置为null

  　3)直到栈为空，遍历结束。

代码重点是，2先访问出栈，然后将5节点呀入栈中

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 InOrder(TreeNode root) {
		if(root==null)return;
		InOrder(root.left);
		System.out.print(root.val+" ");
		InOrder(root.right);
	}
	// 中序遍历 非递归算法
	public static void InOrder2(TreeNode root) {
		if(root==null)return;
		Stack<TreeNode> stk = new Stack<TreeNode>();
		TreeNode p = root;//辅助节点
		stk.add(p);
		while(stk.isEmpty() == false) {
			//只要你有左孩子，就将左孩子压入栈中
			if(p!=null &&  p.left!=null) {
				stk.add(p.left);
				p = p.left;
			}else {
				p = stk.pop();//弹出栈顶节点  左孩子--->根节点
				System.out.print(p.val+" ");//访问
				if(p!=null && p.right!=null) {//如果栈点元素有右孩子的话，将有节点压入栈中
					stk.add(p.right);
					p = p.right;
				}else
					p = null;//p=stk.pop;已经访问过p了，p设置为null
			}
		}
	}
	public static void main(String[] args) {
		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);	
		 
		 InOrder2(root);
	}

}