1

我在用 C++ 为大学项目创建 AVL 树结构时遇到了一些麻烦。到目前为止，我设法制作了简单的添加和删除功能，但问题来了！我在这里看到了一些代码，其中直接在 add 函数中进行检查，但我不想复制代码。有没有办法创建一个单独的函数来完成这个工作？我不能做的事情是如何让程序了解我必须去哪个轮换情况。

到目前为止，我的代码是这样的：

struct node
{
    unsigned int target;
    struct node *left;
    struct node *right;
};

int ClassInvertedIndex::Height(struct node *toCheck)
{
    int left, right;

    if(toCheck == NULL)
        return 0;
    left = Height(toCheck->left);
    right = Height(toCheck->right);
    if(left > right)
        return left+1;
    else
        return right+1;
}

void ClassInvertedIndex::maintainAVL(struct node *toCheck)
{
    if(Height(toCheck->left)-Height(toCheck->right) == 2);  //Left subtree problem.

    if(Height(toCheck->right)-Height(toCheck->left) == 2);  //Right subtree problem.
}


bool ClassInvertedIndex::addNode(unsigned int x)
{
    return insideAdd(data, x);
}



bool ClassInvertedIndex::insideAdd(struct node *toAdd, unsigned int x)
{
    if(toAdd == NULL)
    {
        toAdd = new struct node;

        if(toAdd == NULL)
        {
            cout << "Could not allocate memory.";
            return false;
        }

        toAdd->left = NULL;
        toAdd->right = NULL;
        toAdd->target = x;
        return true;
    }

    else if(x < toAdd->target)
    {
        bool result;
        result = insideAdd(toAdd->left, x);
        if(result)
        {
            //maintainAVL(toAdd);
        }
        return result;
    }

    else if(x > toAdd->target)
    {
        bool result;
        result = insideAdd(toAdd->right, x);
        if(result)
        {
            //maintainAVL(toAdd);
        }
        return result;
    }

    else       //x already exists.
    {
        return false;
    }
}

那么，在maintainAVL 方法中，我应该怎么做才能决定我需要什么轮换呢？

4

0 回答 0