二叉树的下一个节点

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

我的代码：

class Solution {
public:
    vector<TreeLinkNode*> vec;
    void OrderTrave(TreeLinkNode* pRoot)
    {
        OrderTrave(pRoot->left);
        vec.push_back(pRoot);
        OrderTrave(pRoot->right);
    }
    TreeLinkNode* GetNext(TreeLinkNode* pNode)
    {
        if(!pNode) return NULL;
        TreeLinkNode* pRoot = pNode;
        while(pRoot->next)
            pRoot = pRoot->next;
        OrderTrave(pRoot);
        for(int i = 0; i < vec.size()-1; i++)
        {
            if(vec[i] == pNode)
                return vec[i+1];
        }
        return NULL;
    }
};

借鉴后的代码：通过

思路：如果节点为NULL，返回NULL

1.如果节点有右子树，那么一直遍历，找到它的最左面的节点

2.节点没有右子树，那么就要向上找。找到他的父节点，看是不是父节点的左孩子，是则返回父节点，不是则将父节点置为当前节点，重复刚才的步骤判断。

补充：如果节点不是父节点的左孩子，那么就继续向上找，直到为父节点的左孩子。到根还没找得到，就是NULL了

class Solution {
public:
    TreeLinkNode* GetNext(TreeLinkNode* pNode)
    {
        if(!pNode) return NULL;
        if(pNode->right)
        {
            pNode = pNode->right;
            while(pNode->left)
                pNode = pNode->left;
            return pNode;
        }
        while(pNode->next)
        {
            TreeLinkNode* pRoot = pNode->next;
            if(pRoot->left == pNode)
            {
                return pRoot;
            }
            pNode = pNode->next;
        }
        return NULL;
    }
};