题目描述
给定一个二叉树和其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。注意,树中的结点不仅包含左右子结点,同时包含指向父结点的指针。
/*
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.右节点不为空时,取右子树中的最左子节点。以右节点为基准,向左子节点遍历,最左边的子节点即为所找节点。
3.节点没有右子树时,若此节点是其父节点的左子节点,父节点即为所找节点。若此节点是其父节点的右子节点,则沿父结点指针向上遍历,直到找到一个节点是其父节点的左子节点,父节点即为所找节点。
*/
class Solution {
public:
TreeLinkNode* GetNext(TreeLinkNode* pNode)
{
if (pNode == nullptr) //树为空,返回
return nullptr;
TreeLinkNode* nextNode = nullptr; //要求的中序遍历的下一个节点
if (pNode->right != nullptr) //右节点不为空时,取右子树中的最左子节点
{
TreeLinkNode* rightNode = pNode->right; //以右节点为基准
while(rightNode ->left != nullptr) //右节点的左子节点不为空时,向左走,
rightNode =rightNode ->left;
nextNode = rightNode; //最左子节点
}
else if(pNode->next != nullptr) //没有右子树时,父节点不为空
{
TreeLinkNode* currentNode = pNode;
TreeLinkNode* parentNode = pNode->next;
while(parentNode != nullptr && currentNode ==parentNode->right) //节点的父节点不为空,节点是父节点的右子节点时,沿着父节点的指针向上走,直到找到一个节点自己是它父节点的左子节点
{
currentNode = parentNode;
parentNode = parentNode->next;
}
nextNode = parentNode; //父节点就是要找的下一节点
}
return nextNode;
}
};