剑指offer（C++）——二叉树的下一个结点

题目描述

struct TreeLinkNode {
int val;
struct TreeLinkNode *left;
struct TreeLinkNode *right;
struct TreeLinkNode *next;
TreeLinkNode(int x) :val(x), left(NULL), right(NULL), next(NULL) {}
};

/*思路：分两种情况考虑：
（1）当前结点有右子树时：只要从当前结点的右子结点开始，沿着左指针一直寻找下去，最终结点就是当前结点的下一个结点
（2）当前结点没有右子树：a、如果当前结点是其父节点的左子结点，则下一个结点就是其父节点；
b、如果当前结点是其父节点的左子结点，就沿着指向父节点的指针一直向上遍历，直到找到一个结点是其父节点的左子结点为止，这个父节点就是下一个结点*/
class Solution {
public:
TreeLinkNode* GetNext(TreeLinkNode* pNode)
{
if (pNode == NULL)
return NULL;
TreeLinkNode* pNext = NULL;

//如果当前结点有右子树，那么当前结点的下一个结点为其右子树中最左结点
if (pNode->right != NULL)
{
TreeLinkNode* pRight = pNode->right;
while (pRight->left != NULL)               //沿右结点的指向左子结点指针一直找下去
pRight = pRight->left;
pNext = pRight;
}
else if(pNode->next!=NULL)
{
TreeLinkNode* pCurrent = pNode;                                 //始终指向当前遍历的结点
TreeLinkNode* pParent = pNode->next;                            //始终指向父节点
while (pParent != NULL&&pCurrent == pParent->right)             //沿着父指针一直向上遍历
{
pCurrent = pParent;
pParent = pParent->next;
}
pNext = pParent;                 //父节点就是下一个结点（或者父节点为空，说明没有下一个结点）
}
return pNext;
}
};