题目描述
给定一个二叉树和其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。注意,树中的结点不仅包含左右子结点,同时包含指向父结点的指针。
思路如下:
定义的结构体中next指向的是父结点
中序遍历是:左子树 - 根 - 右子树
因此某个节点的下一个节点有下面几种情况:
1. 当前节点的右子树不空,则下一个节点是其右子树中序遍历的第一个节点 (沿着右子树一直向左找第一个叶子节点)
2. 如果右子树为空,又分两种情况
1> 如果此节点是其父节点的左孩子,则其父节点就是下一个节点
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:
//中序遍历: 左子树 - 根 - 右子树
//因此一个节点的下一个节点有下面几种情况:
//1. 此节点的右子树不空 则下一个节点是右子树中序遍历的第一个节点(也就是沿着右子树的左子树一直向左)
//2. 如果右子树为空
// 如果此节点是其父节点的左孩子 则其父节点就是下一个节点
// 如果此节点是其父节点的右孩子 则此节点已经是其父子树的最后一个节点了 需要找到其父节点的下一个节点
// 因此循环判断 找到第一个子节点是父节点的左子树的情况 此时 父节点即为下一个节点
TreeLinkNode* get_inorder_nextnode(TreeLinkNode* pNode){
if(NULL == pNode){
return NULL;
}
if(pNode->right){ //右子树不空的情况
TreeLinkNode* node = pNode->right;
while(node->left){ //找到右子树中序遍历的第一个节点 一直向左 直到第一个叶子节点
node = node->left;
}
return node;
}
else{ //右子树为空的情况
TreeLinkNode* subnode = pNode;
TreeLinkNode* parnode = pNode->next;
while(parnode){
if(subnode == parnode->left){ //如果当前节点是其父节点的左子树 则父节点就是下一个节点
return parnode;
}
else{ //如果当前节点是其父节点的右子树 则循环判断父节点的父节点
subnode = parnode;
parnode = parnode->next;
}
}
}
return NULL;
}
TreeLinkNode* GetNext(TreeLinkNode* pNode)
{
return get_inorder_nextnode(pNode);
}
};