二叉树的下一个结点

题目描述

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

解题思路

先判断这个结点是否为空，不为空就走下一步，判断这个结点是否有右子树，如果有右子树的话找出右子树最左的结点，那么这样就是这个结点的下一个中序遍历的结点了，如果没有右子树，那么找到该节点的的父节点，判断该节点的父节点的左子树是否为这个结点，不是的话向上找，找到为止。可能单凭文字难解析清楚，结合代码更好理解。

/*
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;
        }else{
            while(pNode.next != null){
                if(pNode == pNode.next.left) return pNode.next;
                pNode = pNode.next;
            }
            return null;
        }
    }
}

以上就是这道题的解法。