日常刷题-二叉树的下一个节点

package com.daily.daily20210203;

/**
 * @Project: dailyCode
 * @Site: http://www.zhao1iang.club/
 * @Copyright: ©CodeLamp
 * @Author: zhaoliang
 * @Create: 2021-02-03 16:21
 * @Desc： 二叉树的下一个节点
 **/
public class GetNext {
    class TreeLinkNode{
        int val;
        TreeLinkNode left;
        TreeLinkNode right;
        TreeLinkNode next;
        TreeLinkNode(int x){
            this.val = x;
        }
    }
    public TreeLinkNode getNext(TreeLinkNode pNode){
        //给定一个二叉树和其中的一个结点，请找出中序遍历顺序的下一个结点并且返回。注意，树中的结点不仅包含左右子结点，同时包含指向父结点的指针。
        //如果右子树不为空，下一个节点为右子树的最左节点
        if(pNode.right!=null){
            TreeLinkNode node = pNode.right;
            while (node.left!=null){
                node = node.left;
            }
            return node;
        }else {
            while(pNode.next!=null){
                TreeLinkNode parent = pNode.next;
               if (parent.left == pNode){
                   return parent;
               }
               pNode = pNode.next;
            }
        }
        return null;
    }
}