二叉树的下一个结点

最新推荐文章于 2021-03-17 10:27:35 发布

weixin_34072159

最新推荐文章于 2021-03-17 10:27:35 发布

阅读量71

点赞数

CC 4.0 BY-SA版权

文章标签：数据结构与算法

原文链接：http://www.cnblogs.com/rookieJW/p/9298389.html

本文介绍了一种算法，用于找到给定二叉树中某个结点在中序遍历顺序下的下一个结点。具体实现分为三种情况讨论：有右孩子的结点、无右孩子但为父结点左孩子的结点、无右孩子且为父结点右孩子的结点。

题目描述

给定一个二叉树和其中的一个结点，请找出中序遍历顺序的下一个结点并且返回。注意，树中的结点不仅包含左右子结点，同时包含指向父结点的指针。

思路：

1.如果该结点有右孩子那么下一个结点一定对应着其右孩子最左边的结点。

2.如果该结点没有右孩子，但该结点是其父亲结点的左孩子，那么下一个结点指向其父亲结点

3.如果该结点没有右孩子，且该结点是其父亲结点的右孩子，那么一直往上层遍历直到找到一个是它父结点左结点的结点。

/*
public class TreeLinkNode {
    int val;
    TreeLinkNode left = null;
    TreeLinkNode right = null;
    TreeLinkNode next = null;//指向父亲结点

    TreeLinkNode(int val) {
        this.val = val;
    }
}
*/
public class Solution {
    public TreeLinkNode GetNext(TreeLinkNode pNode)
    {
        //判空
        if(pNode==null){
            return null;
        }
        TreeLinkNode nextNode=null;
        //如果结点还有右子树下一个结点一定在其右子树最左边
        if(pNode.right!=null){
            TreeLinkNode tempNode=pNode.right;
            while(tempNode.left!=null){
                tempNode=tempNode.left;
            }
            nextNode=tempNode;
        }else if(pNode.next!=null){
            TreeLinkNode currentNode=pNode;
            TreeLinkNode parentNode=pNode.next;
            while(parentNode!=null && parentNode.right==currentNode){
                //往上层走
                currentNode=parentNode;
                parentNode=parentNode.next;
            }
            nextNode=parentNode;
        }
        return nextNode;
    }
}

转载于:https://www.cnblogs.com/rookieJW/p/9298389.html