二叉查找树的前驱后继结点:
首先得说明一下什么是前驱?什么是后继
前驱结点:结点值value小于查找结点值search_value的集合中最大值的结点
后继结点:结点值value大于查找结点值search_value的集合中最小值的结点
首先开始定义树结点:
struct TreeNode
{
int val;
TreeNode *left;
TreeNode *right;
TreeNode(int x):val(x), left(NULL), right(NULL);
};
这个时候没图没真相,给个二叉查找树的图看一下:
前驱节点
- 若一个节点有左子树,那么该节点的前驱节点是其左子树中val值最大的节点
- 若一个节点没有左子树,那么判断该节点和其父节点的关系
2.1 若该节点是其父节点的右边孩子,那么该节点的前驱结点即为其父节点。
2.2 若该节点是其父节点的左边孩子,那么需要沿着其父亲节点一直向树的顶端寻找,直到找到一个节点P,P节点是其父节点Q的右边孩子(可参考例子2的前驱结点是1),那么Q就是该节点的前驱节点(按参考例子P结点为5,Q结点为1,P结点是Q结点的右边的孩子),所以Q1就是2结点的前驱节点
后继节点(类比前驱结点)
- 若一个节点有右子树,那么该节点的后继节点是其右子树中val值最小的节点
- 若一个节点没有右子树,那么判断该节点和其父节点的关系
2.1 若该节点是其父节点的左边孩子,那么该节点的后继结点即为其父节点。
2.2 若该节点是其父节点的右边孩子,那么需要沿着其父亲节点一直向树的顶端寻找,直到找到一个节点P,P节点是其父节点Q的左边孩子(可参考例子2的前驱结点是1),那么Q就是该节点的后继节点
下面进行实现的过程
求前驱结点:
TreeNode* getRightNode(TreeNode* root)
{
if (root == NULL)
{
return NULL;
}
while(root->right != NULL)
{
root = root->right; //这个是什么鬼?你没有猜错,我们要找的是前驱节点,
//对于左子树,找左子树的最大值就可以用到这个函数了
}
return root;
}
TreeNode* getPNode(TreeNOde* root, int value, TreeNode*& parent,TreeNode*& firstRParen)
{
while (root)
{
if (root->val == value)
return root;//首先要找到给出节点的位置
//然后记录下父亲节点
parent = root;
if (root->val > value)
{
root = root->left;
}
else
{
firstRParent = root; //出现了右拐点,好像有点开心耶
root = root->right;
}
}
return NULL;
}
//开始搞主函数了
TreeNode* getPreNode(TreeNode* root, int value)//前驱函数
{
if (root)
{
TreeNode* parent = NULL;
TreeNode* firstPRarent = NULL;
TreeNode* node = getPNode(root, value, parent, firstRParent);
if (node == NULL)
return node;//
if (node->left) //高兴吧,有左子树
{
return getRightNode(node->right);
}
if (parent == NULL || (parent && (NULL == firstRParent)))
{
return NULL;//没有前驱节点
}
if (node == parent->right)//右边的孩子,我也是挺高兴的,parent就是他的前驱
{
return parent;
}
else //左边的孩子
{
return firstRParent;
}
}
return root;
}
下面进行实现的过程
求后继结点:
TreeNode* getLightNode(TreeNode* root)
{
if (root == NULL)
{
return NULL;
}
while(root->left != NULL)
{
root = root->left; //这个是什么鬼?你没有猜错,我们要找的是后继节点,
//对于右子树,找右子树的最小值就可以用到这个函数了
}
return root;
}
TreeNode* getNode(TreeNOde* root, int value, TreeNode*& parent,TreeNode*& firstlParen)
{
while (root)
{
if (root->val == value)
return root;//首先要找到给出节点的位置
//然后记录下父亲节点
parent = root; //设置父亲节点
if (root->val > value)
{
root = root->right;
}
else
{
firstPRarent = root; //出现了左拐点,好像有点开心耶
root = root->left;
}
}
return NULL;
}
//开始搞主函数了
TreeNode* getNextNode(TreeNode* root, int value)//后继函数
{
if (root)
{
TreeNode* parent = NULL;
TreeNode* firstlRarent = NULL;
TreeNode* node = getPNode(root, value, parent, firstRParent);
if (node == NULL)
return node;//
if (node->right) //高兴吧,有右子树
{
return getLeftNode(node->right);
}
if (parent == NULL || (parent && (NULL == firstlParent)))
{
return NULL;//没有后继节点
}
if (node == parent->left)//右边的孩子,我也是挺高兴的,parent就是他的前驱
{
return parent;
}
else //没有右子树,
{
return firstlParent;
}
}
return root;
}