题目:
给定一个二叉树和其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。注意,树中的结点不仅包含左右子结点,同时包含指向父结点的指针。
分析:
struct TreeLinkNode {
int val;
struct TreeLinkNode *left;
struct TreeLinkNode *right;
struct TreeLinkNode *next;
TreeLinkNode(int x) :val(x), left(NULL), right(NULL), next(NULL) {
}
};
TreeLinkNode* GetNext(TreeLinkNode* pNode){}
题目的意思是 next 指的该结点的父节点,相当于 parent 指针,按照以往的题套路应该是给这棵树的根节点,以及想要查询的结点信息,但是此题,它没有,根节点需要我们进行循环遍历得到,pNode就是想要查询的结点信息。
结论:
由于是中序遍历,根据中序遍历规则可以得出
1)当pNode是父节点的左孩子时,返回 pNode ->next 即父节点
2)当 PNode 是父节点的右孩子时,返回右孩子的最左结点,否则循环向上遍历,重复判断。
3)其他情况返回NULL
TreeLinkNode* GetNext(TreeLinkNode* pNode)
{
if(pNode == NULL ) return NULL;
if(pNode ->right != NULL)
{
pNode=pNode ->right;
while(pNode ->left != NULL)
{
pNode =pNode->left;
}
return pNode;
}
while(pNode->next != NULL)
{
TreeLinkNode* root=pNode->next;
if(root->left == pNode)
{
return root;
}
pNode = pNode->next;
}
return NULL;
}