【剑指java版】二叉树的下一个节点

文章来源：
个人博客 翔仔年轻有力量：二叉树的下一个节点

题目描述

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

1. 给定一个二叉树和一个节点
2. 找到并返回该节点在中序遍历中的下一个节点
3. 节点类初始化如下：

public class TreeLinkNode {
    int val;
    TreeLinkNode left = null;
    TreeLinkNode right = null;
    //next指针指向的是父节点
    TreeLinkNode next = null;

    TreeLinkNode(int val) {
        this.val = val;
    }
}

---

思路方法

1. 首先判断该节点是否有右子树，如有，下一个节点则为其右子树的最左子节点。
2. 没有右子树的情况下，有两个可能：该节点是它父亲节点的左子节点或右子节点。
   • 当为它父亲节点的左子节点时，下一个节点为其父亲节点。
   • 若为它父亲节点的右子节点时，则代表它父亲节点作为根节点的子树已经遍历完成，需要向上遍历（或者可以说该节点和父节点这一小部分已经遍历，需要向上“扩充操作域”，去寻找原操作域是扩充后操作域的左子树的情况），下一个节点就是操作域为左子树时的父节点。具体图例如下：
     

参考代码

public class Solution {
    public TreeLinkNode GetNext(TreeLinkNode pNode)
    {
        //空二叉树，退出
        if(pNode == null){
            return null;
        }
        //存在右子树
        if(pNode.right != null){
            TreeLinkNode node = pNode.right;
            //通过while循环找到右子树的最左子节点
            while(node.left != null){
                node = node.left;
            }
            return node;
        }
        //循环的边界为存在父节点，即如果向上遍历根节点时，其没有父节点，遍历结束
        while(pNode.next != null){
            //向上遍历，扩充操作域
            TreeLinkNode root = pNode.next;
            //找到原本子树是扩充后子树的左子树
            if(pNode == root.left)
                return root;
            pNode = root;
        }
        return null;
    }
}