二叉树的下一个节点

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

思路：若节点右孩子存在，则设置一个指针从该节点的右孩子出发，一直沿着指向左子结点的指针找到的叶子节点即为下一个节点；若节点不是根节点且该节点是其父节点的左孩子，则返回父节点；否则继续向上遍历其父节点的父节点，重复之前的判断，返回结果

/*
 2 public class TreeLinkNode {
 3     int val;
 4     TreeLinkNode left = null;   //左子节点
 5     TreeLinkNode right = null;  //右子节点
 6     TreeLinkNode next = null;   //父节点
 7 
 8     TreeLinkNode(int val) {
 9         this.val = val;
10     }
11 }
12 */
13 public class Solution {
14     public TreeLinkNode GetNext(TreeLinkNode pNode)
15     {
16         if(pNode==null){
17             return null;
18         }
19         TreeLinkNode pNext=null;
20         if(pNode.right!=null){    //一个节点有右子树
21             TreeLinkNode pRight=pNode.right;
22             while(pRight.left!=null){
23                 pRight=pRight.left;
24             }
25             pNext=pRight;  //下一个节点是该节点右子树中的最左子节点
26         }else if(pNode.next!=null){  //一个节点没有右子树，但是有父节点
27             TreeLinkNode pCur=pNode;
28             TreeLinkNode parent=pNode.next;
29             while(parent!=null&&pCur!=parent.left){ //该节点是它父节点的右子节点
30                 pCur=parent;
31                 parent=pCur.next;
32             }
33             pNext=parent;  //该节点是它父节点的左子节点，则下一个节点就是它的父节点
34         }
35         return pNext;
36     }
37 }

或者

public TreeLinkNode GetNext(TreeLinkNode pNode) {
    if (pNode == null)
        return null;
    if (pNode.right != null) {
        pNode = pNode.right;
        while (pNode.left != null) {
            pNode = pNode.left;
        }
        return pNode;
    }

    while (pNode.next != null) {
        // 找第一个当前节点是父节点左孩子的节点
        if (pNode.next.left == pNode)
            return pNode.next;
        pNode = pNode.next;
    }
    return null;
}