中序遍历--递归和非递归(java版)

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

对于任一结点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 stk = new Stack();
        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);
    }
 
}

你可能感兴趣的:(中序遍历--递归和非递归(java版))