题目描述:
给定一个二叉树和其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。注意,树中的结点不仅包含左右子结点,同时包含指向父结点的指针。
1.该节点有右子树,则中序遍历的下一个节点是右子树的最左的左孩子;
2.该节点只有左子树,
(1)则如果该节点是根节点,没有下一个节点,return NULL;
(2)如果不是根节点,判断该节点是不是父节点的左孩子,是 则下一个节点就是父节点,不是 则将节点指向该节点的父节点,向上回宿,直至找到跟节点
/*
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 -> right)
{
pNode = pNode -> right; //p指向右子树的根节点
while(pNode -> left) pNode = pNode -> left; //存在左子树,直到找到最左边的左孩子
return pNode;
}
//右子树不存在 只有左子树
while(pNode -> next) //next指的是父节点
{
//p不是根节点
if(pNode == pNode -> next -> left) //如果该节点是其父节点的左节点,则其父节点就是下一个节点
return pNode -> next;
pNode = pNode -> next; //如果不是,如D,回溯到C-F,直到找到根节点。
}
return NULL;//p没有父节点,则p是根节点,NULL,没有下一个节点
}
};