二叉树的下一个节点

题目

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

思路

根据给出的节点的父节点指针一直循环往上找可以找到最终的根节点，然后对根节点进行中序遍历，当遍历到当前元素的下一个就是需要的节点。

参考代码

/*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 (!pNode) return pNode;        TreeLinkNode* p = pNode;        while (p->next != p && p->next != nullptr)        {            p = p->next;        }        stack<TreeLinkNode*> st;        bool find = false;        while (p || !st.empty())        {            while (p)            {                st.push(p);                p = p->left;            }            TreeLinkNode* top = st.top();            st.pop();            if (find)            {                p = top;                break;            }            else if (top == pNode)            {                find = true;            }            p = top->right;        }        return p;    }};