c++实现二叉查找树示例

复制代码 代码如下:

#include
#include
#include
#include
#include

using namespace std;

const int M = 10000;

//定义数据节点
class dNode{
public:
 string name;
 int age;
 bool sex;
 dNode(){
  age = 0;
  name = "no name";
  sex = true;//nan
 }
 dNode(string name, int age, bool sex):name(name), age(age), sex(sex){}
 //打印节点
 void show(){
  cout << "name: " << this->name << endl;
  cout << "age: " << this->age << endl;
  cout << "sex: " << this->sex << endl;
  cout << "******************************" << endl;
 }
 //重载赋值符号
 bool operator = (const dNode &d){
  this->age = d.age;
  this->name = d.name;
  this->sex = d.sex;
 }
 //重载相等符号
 bool operator == (const dNode &d){
  return name == d.name && age == d.age && sex == sex;
 }
 //按照年龄重载大于符号
 bool operator > (const dNode &d){
  return age > d.age;
 }
 //按照年龄重载小于符号
 bool operator < (const dNode &d){
  return age < d.age;
 }

};
//定义二叉查找树的节点
//这里规定树中没有重复节点，这里需要对一个节点记录出现多少次
class bstNode{
public: 
 bstNode *left;
 bstNode *right;
 bstNode *parent; //执行父亲，便于向上访问，如果数据量大，并且向上找的使用率不大就不要来减少空间
 dNode data;  //该节点在树中出现的次数
 int count;
 bstNode(){
  left = right = parent = NULL;
  count = 1;
 }
};
//定义二叉树
class bst{
private:
 //清理整棵树
 //注意，一定一定要后续遍历的方法清理
 void destory(bstNode *cur){
  if(NULL == cur){
   return;
  }
  cout << "clearing" << endl;
  destory(cur->left);
  destory(cur->right);
  delete cur; //后续清理
 }
 //真正的删除节点
 void _del(bstNode * cur, bstNode *delNode);
public:
 bstNode * root = NULL;
 bst(){
  root = NULL;
 }
 //插入,返回值是便于构造parent关系
 bstNode * insert(bstNode *& cur, dNode data);
 //搜索
 bstNode * search(bstNode * cur, dNode data);
 //先序遍历
 void pre_raversal(bstNode *cur);
 //返回cur为根的节点的最小值
 bstNode * minNode(bstNode *cur);
 //得到cur节点的后继
 bstNode * succNode(bstNode *cur);
 //删除节点，如果count大于1就将count - 1，如果count==1就清除该节点,返回清除的节点的地址
 bstNode * del(bstNode *cur, dNode data);
 //构造函数对树做清理工作
 virtual ~bst(){
  cout << "###start clear###" << endl;
  this->destory(root);
  cout << "###clear ok###" << endl;
 }

};
bstNode * bst::insert(bstNode *& cur, dNode data){
 if(NULL == cur){
  bstNode * newNode = new bstNode();
  newNode->data = data;
  cur = newNode;
  return cur;
 }else if(cur->data == data){
  cur->count++;
 }else if(cur->data > data){
  bst::insert(cur->left, data)->parent = cur;
 }else if(cur->data < data){
  bst::insert(cur->right, data)->parent = cur;
 }
}
bstNode * bst::search(bstNode *cur, dNode data){
 if(NULL == cur){
  return NULL;
 }else if(cur->data == data){
  return cur;
 }else if(cur->data > data){
  return cur->left;
 }else if(cur->data < data){
  return cur->right;
 }
}
void bst::pre_raversal(bstNode *cur){
 if(NULL == cur)
  return;
 bst::pre_raversal(cur->left);
 cout << "count: " << cur->count << endl;
 cur->data.show();
 bst::pre_raversal(cur->right);
}
bstNode * bst::minNode(bstNode *cur){
 if(NULL == cur){
  return NULL; //如果根节点是空，就返回空
 }else{
  if(NULL != cur->left){
   return minNode(cur->left);
  }
 }
}

bstNode * bst::succNode(bstNode *cur){
 if(NULL != cur->right){
  return minNode(cur);
 }
 bstNode * parentNode = cur->parent;
 while(NULL != parentNode && parentNode->right == cur){
  cur = parentNode;
  parentNode = parentNode->parent;
 }
 return parentNode;
}

void bst::_del(bstNode * cur, bstNode *delNode){
 if(NULL == delNode->left || NULL == delNode->right){
  //待续
 }
}

bstNode * bst::del(bstNode *cur, dNode data){
 //先找到需要删除的节点
 bstNode * delNode = this->search(cur, data);
 if(NULL == delNode) //没有找到该节点，无需删除
  return NULL;
 if(delNode->count == 1){
  _del(this->root, delNode);
 }else{
  delNode->count--;
 }
}

int main(){
 bst *root = new bst();
 //构造50个人， 重复的虽然在树中不会重复插入，但是会被计数
 int num = 50;
 for(int i = 0; i < num; i++){
  dNode * newData = new dNode("Luo", rand() % 15, rand() % 2);
  root->insert(root->root, *newData);
 }
 //前序遍历
 root->pre_raversal(root->root);

 bstNode *searchNode = root->search(root->root, *new dNode("Luo", 3, 1));
 cout << "#######search a Node ##########" << endl;
 if(NULL == searchNode){
  cout << "没有找到该节点" << endl;
 }else{
  cout << "count: " << searchNode->count << endl;
  searchNode->data.show();
 }
 //清理整棵树
 delete root;

 return 0;
}