C ++ - 创建二进制搜索树类,一个节点在旋转后消失

时间:2011-10-31 23:00:37

标签: c++ rotation binary-tree

我正在尝试创建一个自定义二进制搜索树,除了旋转功能之外,我已经完成了所有工作,这似乎没有移动节点。只有在搜索和找到节点时,以及节点是否有正确的子节点时,才会调用rotate函数。为简单起见,我只会添加其中使用的函数,以缩短它:

#include <iostream>

using namespace std;

template <typename T>
class MRBST {
public:
    MRBST();
    ~MRBST();

    void push(const T &);
    bool search(const T &);
    void PrintPreorder();
private:
    struct Node {
        T data;
        Node *left;
        Node *right;

        Node (const T & theData, Node *lt, Node *rt) 
        : data(theData), left(lt), right(rt) {}
    };
    Node *root;

    void push(const T &, Node * &) const;
    void remove(const T &, Node * &) const;
    Node* findMin(Node *t) const {
        if (t == NULL) {
            return NULL;
        }
        if (t->left == NULL) {
            return t;
        }
        return findMin(t->left);
    }
    void preorder(Node * &);
    bool search(const T &, Node *);
    Node* findNode(const T & x, Node * t) {
        if (t == NULL) {
            return NULL;
        }
        if (x < t->data) {
            return findNode(x, t->left);
        } else if (x > t->data) {
            return findNode(x, t->right);
        } else if (x == t->data) {
            return t;
        }
        return NULL;
    }
    void rotate(Node *);
};

template <typename T>
void MRBST<T>::PrintPreorder() {
    preorder(root);
    cout << endl;
}

template <typename T>
void MRBST<T>::preorder(Node * & t) {
    if (t != NULL) {
        cout << t->data << endl;
        preorder(t->left);
        preorder(t->right);
    }
}

template <typename T>
bool MRBST<T>::search(const T & x) {
    if (search(x, root)) {
        Node *temp = findNode(x, root);
        rotate(temp);
        return true;            
    } else {
        return false;
    }
}

template <typename T>
void MRBST<T>::rotate(Node * k1) {
    if (k1->right == NULL) {
        return;
    } else {
        Node *temp = k1->right;
        k1->right = temp->left;
        temp->left = k1;
    }
}


template <typename T>
bool MRBST<T>::search(const T & x, Node *t) {
    if (t == NULL) {
        return false;
    } else if (x < t->data) {
        return  search(x, t->left);
    } else if (x > t->data) {
        return search(x, t->right);
    } else {
        return true;
    }
}

我有一个简单的测试文件,只是在树上添加了一些数字,然后进行搜索,然后在Preordering中打印出来。

#include <iostream>
#include "MRBST.h"

using namespace std;

int main() {
    MRBST<int> binaryTree;
    binaryTree.push(5);
    binaryTree.push(20);
    binaryTree.push(3);
    binaryTree.push(3);
    binaryTree.push(4);
    binaryTree.push(22);
    binaryTree.push(17);
    binaryTree.push(18);
    binaryTree.push(8);
    binaryTree.push(9);
    binaryTree.push(1);
    binaryTree.PrintPreorder();
    cout << endl;
    binaryTree.search(17);
    binaryTree.PrintPreorder();
    cout << endl;

}

输出看起来像这样:

5 3 1 4 20 17 8 9 18 22

5 3 1 4 20 17 8 9 22

如果有人能够对旋转功能出现问题有所了解,那将是一个很大的帮助。

2 个答案:

答案 0 :(得分:1)

为什么你在搜索时rotate?它应该是只读的。

你因此失去了节点。

查看您的rotate

    Node *temp = k1->right;
    k1->right = temp->left;
    temp->left = k1;

假设k1 = x有right = y而left = z,一步一步看:

    Node *temp = k1->right;

temp = k1-&gt; right = y,k1 = x,k1-&gt; left = z

    k1->right = temp->left;

k1-&gt; right = ?, k1-&gt; left = z,temp = y

    temp->left = k1;

k1-&gt; right = ?, k1-&gt; left = z,temp-&gt; left = x。

现在 - Y去哪了?你失去了它。

答案 1 :(得分:0)

仔细查看你的rotate()函数,看看它是否符合你的要求。