JZ8--- 二叉树的下一个结点(Java)

思路展示

//如果一个节点有右子树（那么它的下一个节点就是他的右子树中的最左子节点）
    //如果一个节点没有右子树
        //如果此节点没有父节点，则其父节点为null
       //如果此节点是他父节点的左子节点（那么它的下一个节点就是他的父节点）
       //如果此节点是他父节点的左子节点（找到一个是它父节点的左子节点的节点）

注意：如果一个节点既没有右子树，并且它还是它父节点的右子节点，那么这种情形就比较复杂。我们可以沿着指向父节点的指针一直向上遍历，直到找到一个是它父节点的左子节点的节点。如果这样的节点存在，那么这个节点的父节点就是我们要找的下一个节点。
为了找到图中节点i的下一个节点，我们沿着指向父节点的指针向上遍历，先到达节点e。由于节点e是它父节点b的右节点，我们继续向上遍历到达节点b。节点b是它父节点a的左子节点，因此节点b的父节点a就是节点i的下一个节点。

示例代码

import java.util.*;
/*
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) {
        //如果一个节点有右子树（那么它的下一个节点就是他的右子树中的最左子节点）
        //如果一个节点没有右子树
            //如果此节点没有父节点，则其父节点为null
           //如果此节点是他父节点的左子节点（那么它的下一个节点就是他的父节点）
           //如果此节点是他父节点的左子节点（找到一个是它父节点的左子节点的节点）

        if(pNode==null){
            return null;
        }
        TreeLinkNode nextNode=null;
        if(pNode.right!=null){
            TreeLinkNode curNode=pNode.right;
            while(curNode.left!=null){
                curNode=curNode.left;
            }
            return curNode;
        }else{
            if(pNode.next==null){
                return nextNode;
            }else{
                TreeLinkNode curNode=pNode;
                TreeLinkNode parentNode=pNode.next;
                while(parentNode!=null&&parentNode.right==curNode){
                    curNode=parentNode;
                    parentNode=parentNode.next;
                }
                nextNode=parentNode;
            }
        }
        return nextNode;
    }
}

效果展示