题目描述:给定一颗二叉树和其中的一个节点,如何找出中序遍历序列的下一个节点?树中的节点除了有两个分别指向左右子树节点的指针,还有一个指向父节点的指针。
解题思路:这道题目就是要理清二叉树的前、中、后序遍历,然后分情况理清楚下一个节点就行。要举一反三。
- 如果这个节点有右子树,则下一个节点是右子树中最左子节点。
- 如果这个节点没有右子树,且节点是父节点的左子节点,则下一个节点是其父节点。
- 如果这个节点没有右子树,且节点是父节点的右子节点。沿着父节点指针一直向上遍历,直到找到一个节点,他是其父节点的左子节点,则下一个节点就是这个节点的父节点。
/*
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;
}
//如果该节点有右子树,则下一个节点是右子树的最左节点
if(pNode->right!=nullptr)
{
TreeLinkNode* ptr=pNode->right;
while(ptr->left!=nullptr)
{
ptr=ptr->left;
}
return ptr;
}
//如果该节点没有右子树,且该节点是父节点的左子树,则下一个节点是父节点
if(pNode->next!=nullptr&&pNode==pNode->next->left)
{
return pNode->next;
}
//如果该节点既没有右子树,又是父节点的右子树时,沿着父节点指针一直向上遍历,
//直到找到一个节点,他是父节点的左子树节点,则这个节点的父节点就是下一个节点
while(pNode->next!=nullptr&&pNode==pNode->next->right)
{
pNode=pNode->next;
}
return pNode->next;
}
};