레드블랙트리 증감연산자 / C++

증가연산자의 경우에는 마지막 원소다음에는 end iterator를 반환하고 감소연산자의 경우에는 begin iterator를 감소하는 경우에는 assert 처리를 해주었다. 레드블랙트리에서의 삭제의 알고리즘이 복잡해서 더 공부하고 구현예정이다.

 
RedBlackTree::iterator& RedBlackTree::iterator::operator++()
{
    Node* nPtr = this->node;
    assert(nPtr);
    Node* leaf = tree->Leaf;
    // #1 오른쪽 자식이 있음 
    if (nPtr->RightChild != leaf)
    {
        nPtr = nPtr->RightChild;
        while (nPtr->LeftChild != leaf)
        {
            nPtr = nPtr->LeftChild;
        }
        this->node = nPtr;
    }
    else // 부모의 오른쪽 자식이고 오른쪽 자식이 없음
    {
        while (true)
        {
            if (nPtr->Parent == nullptr)
            {
                this->node = nullptr;
                break;
            }
            if (tree->IsLeftChild(nPtr))
            {
                nPtr = nPtr->Parent;
                this->node = nPtr;
                break;
            }    
            nPtr = nPtr->Parent;
        }
    }
 
    return *this;
}
 
RedBlackTree::iterator& RedBlackTree::iterator::operator--()
{
    Node* nPtr = this->node;
    Node* leaf = tree->Leaf;
    // #0 end iterator 인 경우 
    if (nPtr == nullptr)
        return tree->rbegin();
    // #1 왼쪽자식
    else if (nPtr->LeftChild != leaf)
    {
        nPtr = nPtr->LeftChild;
        while (nPtr->RightChild != leaf)
        {
            nPtr = nPtr->RightChild;
        }
        this->node = nPtr;
    }
    else
    {
        while (true)
        {
            if (nPtr->Parent == nullptr)
                assert(nullptr);
            if (tree->IsRightChild(nPtr))
            {
                nPtr = nPtr->Parent;
                this->node = nPtr;
                break;
            }
            nPtr = nPtr->Parent;
        }
    }
 
    return *this;
}