目录
迭代器自增
当前结点存在右子树
当前结点没有存在右子树
迭代器自增完整代码
迭代器自减
迭代器自减代码:
迭代器自增
红黑树的迭代器应该怎么实现呢?现在我模仿大佬的实现逻辑。
我们迭代器最重要是可以允许自增与自减的实现的。
会发现我们遍历的时候我们迭代器是根据红黑树的最左边结点进行遍历数据的。
当前结点存在右子树
虽然看图我们知道下一个我们是遍历1结点,但是我我们并不知道0的右边有没有东西,所以我们需要判断。
Self& operator++()
{
if (_node==nullptr)
{
return *this;
}
if (_node->_right)
{
//下一个就是右子树的最左结点
Node* left = _node->_right;
while (left->_left)
{
left = left->_left;
}
_node = left;
}
else
{
//当前结点不存在右子树的访问
}
return *this;
}
所以我们需要先判断我们的当前位置是否存在右子树,如果没有才能访问上一级结点。
当前结点没有存在右子树
当我们没有当前结点没有右子树就会访问当前结点的上层结点
然后再次it++,当前结点存在右子树,it来都右子树的最左结点(没有2结点没有左子树,那么自己就是最左结点)。
这时我们在it++,当前结点没有右子树返回上级,但是我们的上级结点1已经被访问过了,所以我们需要在当前结点右子树为空的时候返回上级需要判断当前结点是否是上级结点的右子树,如果是迭代的再去上层找,找这一路径里是父节点的左子树位置,然后将该父节点赋值给it。
访问到最后,it会到整棵树的根的父节点nullptr
代码:
Self& operator++()
{
if (_node==nullptr)
{
return *this;
}
if (_node->_right)
{
//下一个就是右子树的最左结点
}
else
{
//右子树为空,找这一层里不是父节点的右的哪个
Node* parent_nf = _node->_parent;
Node* _cur = _node;
while (parent_nf&&parent_nf->_right == _cur)//存在到最后一个结点会访问
//到整棵树的根结点的父结点,就是访问完毕
{
_cur = parent_nf;
parent_nf = parent_nf->_parent;
}
_node = parent_nf;
}
return *this;
}
迭代器自增完整代码
Self& operator++()
{
if (_node == nullptr)
{
return *this;
}
if (_node->_right)
{
//下一个就是右子树的最左结点
Node* left = _node->_right;
while (left->_left)
{
left = left->_left;
}
_node = left;
}
else
{
//右子树为空,找这一层里不是父节点的右的哪个
Node* parent_nf = _node->_parent;
Node* _cur = _node;
while (parent_nf && parent_nf->_right == _cur)
{
_cur = parent_nf;
parent_nf = parent_nf->_parent;
}
_node = parent_nf;
}
return *this;
}
迭代器自减
迭代器自减就是和自增反向的逻辑
迭代器自减代码:
Self& operator--()
{
if (_node->_left)//当前结点左树存在
{
Node* right = _node->_left;
while (right->_right)//选择左子树的最右结点,
//就是迭代器自减的下一个位置
{
right = right->_right;
}
_node = right;//找到左子树的最右节点。
}
else /*if(_node->left==nullptr)*/ //当前结点左子树存在
{
Node* parent_nf = _node->_parent;
Node* _cur = _node;
while (parent_nf && parent_nf->_left == _cur)
//寻找当前位置不是父子子树的左子树。
{
_cur = parent_nf;
parent_nf = parent_nf->_parent;
}
_node = parent_nf;
}
return *this;
}