• 剑指offer:JZ8 二叉树的下一个结点


    JZ8 二叉树的下一个结点

    描述

    给定一个二叉树其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。注意,树中的结点不仅包含左右子结点,同时包含指向父结点的next指针。下图为一棵有9个节点的二叉树。树中从父节点指向子节点的指针用实线表示,从子节点指向父节点的用虚线表示

    示例:
    输入:{8,6,10,5,7,9,11},8
    返回:9
    解析:这个组装传入的子树根节点,其实就是整颗树,中序遍历{5,6,7,8,9,10,11},根节点8的下一个节点就是9,应该返回{9,10,11},后台只打印子树的下一个节点,所以只会打印9,如下图,其实都有指向左右孩子的指针,还有指向父节点的指针,下图没有画出来

    分析

    初拿到这道题,很明显看出是一个二叉树的中序遍历就可以解决的问题,但是仔细看了题目之后,发现,题目会给两个参数,而这两个参数组装成的是一个二叉树节点(可以是其中树的任意节点),这样的话,如果使用普通的遍历方法直接做,就会有问题,需要多一步遍历这个节点的next找到整棵树的根节点,之后使用dfs进行中序遍历得到结果。

    代码

          /**
         * JZ8 二叉树的下一个结点
         * @param pNode
         * @return
         */
        public TreeLinkNode GetNext(TreeLinkNode pNode) {
            //提出题目中要求对比的值
            Integer value = pNode.val;
            //通过不断的寻找next找到整棵树的根节点
            while (pNode.next != null) {
                pNode = pNode.next;
            }
            //结果队列
            Queue<TreeLinkNode> queue = new LinkedList<>();
            //dfs遍历使用的栈,储存没有遍历过的节点用
            Stack<PosNode> treeStack = new Stack<>();
            //根节点
            PosNode rootNode = new PosNode(pNode);
            //根节点加入栈
            treeStack.push(rootNode);
            //如果所有节点都已经遍历过就跳出循环
            while (treeStack.size() > 0) {
                //中序遍历,先遍历左子树
                if (treeStack.peek().node.left != null && treeStack.peek().visitLeft == false) {
                    PosNode posNode = new PosNode(treeStack.peek().node.left);
                    //证明左子树访问过
                    treeStack.peek().visitLeft = true;
                    treeStack.push(posNode);
                }else {
                    //把节点值值存储在结果队列中
                    queue.offer(treeStack.peek().node);
                    //证明左子树访问过
                    treeStack.peek().visitLeft = true;
                    //开始遍历右子树
                    if (treeStack.peek().node.right != null && treeStack.peek().isVisitRight == false) {
                        PosNode posNode = new PosNode(treeStack.peek().node.right);
                        treeStack.peek().isVisitRight = true;
                        treeStack.push(posNode);
                    }else {
                        //如果右子树也遍历过或者为空的话
                        treeStack.peek().isVisitRight = true;
                        //通过出栈寻找左右子树都没有遍历过的节点
                        while (treeStack.size() != 0 && treeStack.peek().isVisitRight == true && treeStack.peek().visitLeft == true)
                            treeStack.pop();
                    }
                }
            }
    
            //存储的中序遍历结果进行值的比较找出结果
            while (queue.size() != 0) {
                if (queue.peek().val != value) {
                    queue.poll();
                }else {
                    queue.poll();
                    return queue.peek();
                }
            }
    
            //找不到的话返回空值
            return null;
        }
    
        /**
         * 保存节点以及记录节点的左右子树是否访问过的表质量
         */
        class PosNode{
            TreeLinkNode node;
            boolean visitLeft = false;
            boolean isVisitRight = false;
    
            PosNode(TreeLinkNode node) {
                this.node = node;
            }
        }
    

    总结

    需要熟练使用代码来遍历二叉树,本人的方法使用内存占用较高,后续可以去除结果队列,直接在遍历中使用标志量记录结果返回。

  • 相关阅读:
    安装vim
    linux输入法
    freemarker 超出字符用省略号表示
    三维数组去重
    php 相同的产品 一个背景色
    php 判断白天黑夜
    PHP-preg_replace过滤字符串代码
    checkbox 更新回显
    php socket编程入门
    命令行执行php
  • 原文地址:https://www.cnblogs.com/bearcanlight/p/15434460.html
Copyright © 2020-2023  润新知