题目描述
给定一个二叉树和其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。注意,树中的结点不仅包含左右子结点,同时包含指向父结点的指针。
/*
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、当pNode为NULL,直接返回NULL
//2、当pNode的右孩子不为空,则返回中序遍历右子树的第一个结点即可
//3、当pNode的右孩子为空时,则代表以pNode为根的子树已经遍历完成
// 返回到父节点,如果pNode是父节点的左孩子,则下一个遍历结点就是
// 父节点,如果pNode是父节点的右孩子,则代表以pNode的父节点为根
// 的子树已经遍历完成,继续向上寻找。(有可能父节点为空,代表整个树
遍历完成,也返回空值)
class Solution {
public:
TreeLinkNode* GetNext(TreeLinkNode* pNode)
{
if(pNode==NULL)return NULL;
if(pNode->right==NULL)
{
if(pNode->next!=NULL&&pNode==pNode->next->left)
return pNode->next;
else if(pNode->next==NULL)return NULL;
else
{
auto ptr=pNode;
while(ptr->next!=NULL&&ptr==ptr->next->right)
{
ptr=ptr->next;
}
return ptr->next;
}
}
else
{
TreeLinkNode* ptr=pNode->right;
while(ptr->left!=NULL)
{
ptr=ptr->left;
}
return ptr;
}
}
};