剑指offer_【57】二叉树的下一个结点

1.题目描述

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

2.解题思路

情况1：该节点有右子树：

         6

     /      \

   3         10

 /   \      /    \

2     5    8      12

中序遍历结果为：2–>3–>5–>6–>8–>10–>12

即它的下一个结点就是它的右子树中最左子结点

情况2.1：该节点无右子树：(为父节点6的左子节点)

        6

     /     \

   3        10

  /        /    \

2         8      12

2–>3–>6–>8–>10–>12

该节点是父节点的左子节点的这种情况比较简单，直接将父节点返回即可

情况2.2：该节点无右子树：(为父节点6的右子节点)

          6

       /    \

     3        10

  /     \     /   \

2       5    8    null

2–>3–>6–>8–>10 –>null

如果是父节点的右子节点的话，需要不断的向上移动，直到对应的节点不是父节点的右节点（即左节点），既然他是父节点的左节点，此时将这个节点父节点返回即可，或者遍历到了根节点，返回null；

3.代码

/*
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){
            pNode = pNode.right;
            while(pNode.left != null){
                pNode = pNode.left;
            }
            return pNode;
        }
        //该节点无右子树，父节点不为空

        while(pNode.next != null){
            //pNode为父节点的左节点为该节点，直接返回父节点
            if(pNode.next.left== pNode){
                return pNode.next;
            }
       //为父节点的右子节点，不断的向上移动，直到对应的节点不是父节点的左子节点                  
       //一直回溯如果遍历到他是父节点的左节点，此时将这个节点父节点返回即可，
       //或者遍历到了根节点，返回null；

            pNode = pNode.next;
        }
        return null;
    }
}