二叉树的下一个结点

Posted 2021-01-11 MarkLeeBYR

题目描述

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

树见书P275

分三种情况：

1. 该节点有右子树，下一个结点就是它右子树的最左节点

2. 该节点没有右子树，而且它是父节点的左子节点，则下一个结点是它的父节点

3. 该节点没有右子树，而且它是父节点的右子节点，就沿着父节点指针向上遍历，直到找到一个节点(是自己父节点的左子节点)，该节点的父节点就是要找的下一个结点

4. 该节点是全树最右结点，所以没有下一个结点了

public class Solution {

    public TreeLinkNode GetNext(TreeLinkNode pNode) {

        if (pNode == null)

            return null;

        if (pNode.right != null) {//情况1

            pNode = pNode.right;

            while (pNode.left != null)

                pNode = pNode.left;

            return pNode;

        }

        while(pNode.next != null){ //包括了情况2和3

            if(pNode.next.left == pNode) 

                return pNode.next;

            pNode = pNode.next;

        }  

        return null;//情况4

    }

}