我正在尝试创建一个自定义二进制搜索树,除了旋转功能之外,我已经完成了所有工作,这似乎没有移动节点。只有在搜索和找到节点时,以及节点是否有正确的子节点时,才会调用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
如果有人能够对旋转功能出现问题有所了解,那将是一个很大的帮助。
答案 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()函数,看看它是否符合你的要求。