二叉树的下一个节点

题目描述

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

 

两种情况

情况1.这种情况下得找自己祖先是哪个隔代祖先的左子树。如果一直是右子树到了根节点，那说明到头了，下一个是null。

情况2.理解:※ 右子树的最左节点。

/*
struct TreeLinkNode {
    int val;
    struct TreeLinkNode *left;
    struct TreeLinkNode *right;
    struct TreeLinkNode *next;
    TreeLinkNode(int x) :val(x), left(NULL), right(NULL), next(NULL) {
        
    }
};
*/
class Solution {
public:
    TreeLinkNode* GetNext(TreeLinkNode* pNode) {
        
        //中序遍历伪代码。
//         if (root == null) return;
//         traverse(root.left);
//         visit(root);
//         traverse(root.right);
        
        //肯定有需要找自己祖先的情况。
        //暴力解法。
//        1.根据给出的结点求出整棵树的根节点
//        2.根据根节点递归求出树的中序遍历，存入vector
//        3.在vector中查找当前结点，则当前结点的下一结点即为所求
        
        //（ 肯定有需要找自己祖先的情况。
        //如果其有右子树的话，肯定是自己的儿子或者孙子或者下一辈。
        //如果其没有右子树，才需要回溯到自己的祖先。如果是自己祖先的左子树，那祖先还没访问过，直接返回祖先节点。
        //（接上句）但是若自己是自家祖先的右子树，那直系祖先已经访问过，得找自己祖先是哪个隔代祖先的左子树。）
        
        //while或另写两个递归函数。/当右子节点为空时，向上递归；当右子节点不为空时，向下递归。
        if(pNode->right==NULL)
        {
            //这种情况下得找自己祖先是哪个隔代祖先的左子树。如果一直是右子树到了根节点，那说明到头了，下一个是null。
            while(pNode->next&&pNode->next->left!=pNode)
            {
                pNode=pNode->next;
            }
            return pNode->next;
        }
        else
        {
            //理解：※ 右子树的最左节点
            pNode=pNode->right;//右子树
            while(pNode->left) pNode=pNode->left;//最左节点。
            return pNode;
        }
        return NULL;
    }
};