二叉树前、中、后序遍历

遍历二叉树

递归方式遍历

public static void pre(Node head){
    if(head == null) return;
    System.out.print(head.value+"-");
    pre(head.left);
    pre(head.right);
}
public static void in(Node head){
    if(head==null) return;
    in(head.left);
    System.out.print(head.value+"-");
    in(head.right);
}
public static void pos(Node head){
    if(head==null) return;
    pos(head.left);
    pos(head.right);
    System.out.print(head.value+"-");
}

没什么好说的,就是打印的时间点不同,先序是先打印头,然后处理左子树,再处理右子树,中序和后序同理。

认识递归序

public static void f(Node head) {
   if (head == null) {
      return;
   }
   f(head.left);
   f(head.right);
}

看上面的三个算法,你会发现主要的代码是相同的,就是打印的时机不同,就能够完成三种遍历,其实这种便利性受益于递归强大的功能,就是递归序

比如这个树,

image-20210220235158198.png

从1开始执行f函数,那么会先进入1节点,然后去左树溜一圈,再返回1节点。然后在去右子树溜一圈,再返回1节点,其他节点同理,可以得到递归序

①->②->④->④->④->②->⑤->⑤->⑤->②->①->③->⑥->⑥->⑥->③->⑦->⑦->⑦->③->①

上面这个就是上面那个树的递归序,每个节点表示传进函数的节点,递归之所以强大就是因为左子树右子树能够带些消息回头,然后头能够对它们两的消息进行一个整合,像是遍历二叉树,让左子树,右子树都带消息回来,若是中序就等左子树都遍历完了回来再打印等等。

像是归并排序就是依赖了递归序的强大,先去把左边排好序,再去右边排好序,最后回来merge使得整体有序,比起O(N^2)的强大之处就是每次比较后它的信息都是留下来的,返回到头的时候进行整合把两边的信息加和后传给更高层的节点。

改写成非递归

递归的本质就是压栈,所以可以从栈的角度来思考怎么改写。

前序

的思想就是,先弹出打印头,然后把右压入,把左压入,这样之后处理的时候就会先处理左边再处理右边,栈就是起到一个逆序的作用。每次都是如此,先头再左最后右,这就是前序的要求。

//口诀:1)弹打印2)如有右,压入右3)如有左,压入左
public static void pre(Node head){
    if(head!=null){
        StringBuilder preOrder = new StringBuilder("preOrder:");
        Stack<Node> stack = new Stack<>();
        stack.push(head);
        while(!stack.isEmpty()){
            head = stack.pop();
            preOrder.append(head.value).append("-");
            //先右
            if(head.right!=null){
                stack.push(head.right);
            }
            //再左
            if(head.left!=null){
                stack.push(head.left);
            }
        }
        preOrder.deleteCharAt(preOrder.length()-1);
        Logger.getGlobal().info(preOrder.toString());
    }
}

中序

中序和前序的区别是不能直接就弹出打印,需要将左子树全部打印完才能打印头,然后处理右子树。思路是,先一直往左窜,不断把左边节点一股脑压入栈,知道压不了了表示这一路最后那个叶子节点已经处理好左子树了,可以弹出那个叶子节点打印它,然后如果它有右节点就压入右节点。

image-20210221003428523.png

这个实现的思路就是一个二叉树可以分解成很多个左子树,然后压入栈的时候呈现一个先头再左的顺序,然后出栈先出左,就是先处理左。

public static void in(Node head) {
    if (head != null) {
        StringBuilder inOrder = new StringBuilder("inOrder:");
        Stack<Node> stack = new Stack<>();
        //中序遍历一个树的流程,这个数的头如果不为空我就将它的左子树全部压入
        //若是碰到头为空了,表示最后一个节点的左子树处理完,就弹出最后压入的节点打印
        //处理它的右子树
        while (head != null || !stack.isEmpty()) {
            if (head != null) {//处理左
                stack.push(head);
                head = head.left;
            } else {//处理右
                head = stack.pop();
                inOrder.append(head.value).append("-");
                head = head.right;
            }
        }
        //基本思路就是先压入[头、左...]这样的顺序,然后没有左了就pop出最后一个左边的节点,它的右如果有就压入
        // ,此时这个左节点的左子树和它自己已经处理过了,然后压入右子树处理符合中序。
        //它的右子树处理完全弹出后再次弹出的就是这个左节点的父,然后这个父又去处理它的右子树,周而复始,完成遍历
        inOrder.deleteCharAt(inOrder.length() - 1);
        Logger.getGlobal().info(inOrder.toString());
    }
}

为什么中序不和前序一样先压头,然后以栈是否为空进入循环呢?

额,这样你进去还不能像前序一样直接弹头打印,你得先peek看一下,然后用这个peek看到的头去判断它的左树是否为空,这样你之前已经处理过节点的左树了,结果现在又来看就会死循环的。上面这个代码的妙处就是这样就是每次处理一个节点的时候都会指向他的右树,也就是说就不会再看回左树了,就避免了死循环。

后序

使用两个栈

//我们已经知道前序,头左右是先处理头然后有右就压入右,再有左压入左
//修改一下就会产生一个处理头右左的序列,但是我们先不进行打印,而是再次入栈进行一个逆序那就是后序了
public static void pos(Node head) {
    if (head != null) {
        StringBuilder posOrder = new StringBuilder("posOrder:");
        Stack<Node> stack1 = new Stack<>();
        Stack<Node> stack2 = new Stack<>();
        stack1.push(head);
        while (!stack1.isEmpty()) {
            head = stack1.pop();
            if (head.left != null) {
                stack1.push(head.left);
            }
            if (head.right != null) {
                stack1.push(head.right);
            }
            stack2.push(head);//弹出来的东西进栈2
        }
        while (!stack2.isEmpty()) {
            posOrder.append(stack2.pop().value).append("-");
        }
        posOrder.deleteCharAt(posOrder.length() - 1);
        Logger.getGlobal().info(posOrder.toString());
    }

}

但是经过前序思路修改的后序需要多一个容器,能不能考虑只用一个容器呢?
可以。仔细研究中序的思路,你会发现中序的思路是先把左处理完后马上处理头,然后处理右,这样右子树处理完之后再次弹出来的就是这次处理的头的父,然后它去处理它的右子树。

能不能,把左处理完之后先不处理头,然后处理右边,等右边也处理完了再处理头呢?要完成这个的关键点是如何区分到底回来头这里的情况,是才完成了左边还是已经完成了右边了。
方案:我整个指针,每次都指向上一次处理的节点,然后回到头的时候看一下那个指针不就知道上一次完成的是左还是右了吗

public static void pos2(Node head) {
    if (head != null) {
        StringBuilder posOrder = new StringBuilder("posOrder:");
        //也就是需要两个指针,这里就复用了head,用head来做那个移动遍历的指针
        Node last = head;//这个指针一开始还没有我们需要的意思,第一次处理了一个节点后它才有用,所以先指向一个不会影响的地方
        //指向空的话可能这个二叉树不是满二叉树,这样就会出现问题,比如一个节点只有左孩子没有右孩子,此时比对right的
        //时候null==null成立,就会不接着处理左孩子了
        Stack<Node> stack = new Stack<>();//栈还是得有的,总得记住还来不及处理的节点
        stack.push(head);
        while (!stack.isEmpty()) {
            head = stack.peek();
            if (head.left != null && last != head.left && last != head.right) {
                //有左,而且上一次处理的不是左而且不是右,表示这个点才开始处理
                stack.push(head.left);
            } else if (head.right != null && last != head.right) {
                //来到这个if表示左肯定是处理过了,只要上一次没处理过右,而且有右,那就处理右去
                stack.push(head.right);
            } else {
                //来到这里表示左右都处理过了,处理它自己,而且将它弹出去
                posOrder.append(stack.pop().value).append("-");
                last = head;
            }
        }
        posOrder.deleteCharAt(posOrder.length() - 1);
        Logger.getGlobal().info(posOrder.toString());
    }
    //这里思路上用的中序的思路,但是实现上确是和前序相似,因为如果这时候用中序类似的代码实现,
    //就是一开始不压入头,这会出现这么个问题,就是遍历开始前需要先压入当前head,如果这个点左右
    //处理过了,你又得给它弹出来,一次遍历完成之后你得给head指向下一个它需要处理的点,可能没有
    //需要处理的点了,你还得判断一下栈是否为空,就是下面的pos3,可以比较一下
}
//这样不好,反复拿出来又放回去,影响效率

public static void pos3(Node head) {
    if (head != null) {
        Stack<Node> stack = new Stack<>();
        StringBuilder posOrder = new StringBuilder("posOrder:");
        Node last = head;
        while (head != null || !stack.isEmpty()) {
            stack.push(head);
            if (head.left != null && last != head.left && last != head.right) {
                head = head.left;
            } else if (head.right != null && last != head.right) {
                head = head.right;
            } else {
                posOrder.append(stack.pop().value).append("-");
                last = head;
                if (stack.isEmpty()) break;
                head = stack.pop();
            }
        }

        posOrder.deleteCharAt(posOrder.length() - 1);
        Logger.getGlobal().info(posOrder.toString());
    }
}

我觉得这个非递归去遍历二叉树如果不是有什么特别的兴趣的话,强记住好的三种实现就好了,我自己琢磨了一天感觉收获不大,写出的代码也没这三种妙。

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 213,864评论 6 494
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 91,175评论 3 387
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 159,401评论 0 349
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 57,170评论 1 286
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 66,276评论 6 385
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,364评论 1 292
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,401评论 3 412
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,179评论 0 269
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,604评论 1 306
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,902评论 2 328
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,070评论 1 341
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,751评论 4 337
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,380评论 3 319
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,077评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,312评论 1 267
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 46,924评论 2 365
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 43,957评论 2 351

推荐阅读更多精彩内容