题目描述
给定一个二叉树和其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。注意,树中的结点不仅包含左右子结点,同时包含指向父结点的指针。
思路:题中给出要求,为中序遍历顺序,即左、根、右;如果给定节点存在右节点,则找到其右节点的最左节点即为下一个节点;如果不存在右节点,则寻找父节点是其父节点的左子节点,即如果还存在下一节点则该节点必须位于整棵树的左子树,若位于右子树则寻找不到上述要求的节点,则返回 n u l l p t r nullptr nullptr
/*
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 == nullptr)
return nullptr;
TreeLinkNode* pNext = nullptr;
//存在右节点,则找到右节点的最左节点
if (pNode->right != nullptr) {
TreeLinkNode* pRight = pNode->right;
while (pRight->left != nullptr) {
pRight = pRight->left;
}
pNext = pRight;
}
//不存在右子节点,则找到节点是其父节点的左子节点;找不到则返回nullptr
else if (pNode->next != nullptr) {
TreeLinkNode* pCurrent = pNode;
TreeLinkNode* pParent = pNode->next;
while (pParent != nullptr && pCurrent == pParent->right) {
pCurrent = pParent;
pParent = pParent->next;
}
pNext = pParent;
}
return pNext;
}
};