剑指Offer——二叉树的下一个结点

题目描述

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

思路

分情况：

1. 当前节点有右子树>>>>>右子树的最左那个节点
2. 当前节点没有右子树：
   2.1当前节点是左节点>>>>>父节点
   2.2当前节点是右节点>>>>>父辈节点中是左节点的父节点

/*
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 pNext = null;
        
        //1.如果当前节点有右子树
        if(pNode.right!=null){
            pNext = pNode.right;
            while(pNext.left!=null){
                pNext = pNext.left;
            }
            return pNext;
        }//2.如果没有右子树
        else if(pNode.next !=null){
            //2.1当前节点是左子树
            if(pNode.next.left == pNode){
                pNext = pNode.next;
            }else{
                //2.2当前节点是右子树
                TreeLinkNode cur = pNode;
                TreeLinkNode par = pNode.next;
                while(par!=null&&cur == par.right){
                    cur = par;
                    par = par.next;
                }
                pNext = par;
            }
            return pNext;
        }
        return null;
    }
}