《剑指Offer》之“二叉树的下一个结点”

题目描述

---

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

代码实现

---

//分成三个大的逻辑块来处理
/*
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;
    }    
​     //当前节点有右节点
        if (pNode.right != null) {
            TreeLinkNode p = pNode.right;
            while (p.left != null) {
                p = p.left;
            }
            return p;
        }
​      //当前节点没有右节点
        else{
​            //判断是否为根节点，根节点就没有必要分左右分支了
             if(pNode.next == null  ){
                return null;
                }
            TreeLinkNode  p = pNode;
            TreeLinkNode  pc = null ;
            while(p.next != null){
                pc = p;
                p = p.next;
            }
            //从根节点划分出来的左分支
            if(p.left == pc){
​               //相对父节点来说在左分支
                if(pNode.next.right == pNode){
                return p;
                }
​               //相对父节点来说在右分支
                else{
                    return pNode.next;
                }
            }
            //从根节点划分出来的右分支
            else{
​                 //相对父节点来说在左分支
                 if(pNode.next.right == pNode){
                return null;
                }
​                  //相对父节点来说在右分支
                else{
                    return pNode.next;
                }
            }
        }
    }
}
​