二叉搜索树的两种实现（数组模拟，STL）

书上实现：

二叉搜索数的特点：高效实现 插入一个数值，查询是否包含某个数值，删除某一个数值。

所有的节点都满足左子树上的所有节点都比自己的小，而右子树上的所有节点都比自己大的特点。

查询：如果当前数值等于根节点返回true，比根节点小，就往左儿子走，否则往右儿子走。

插入：按照查找数值的方法去找其所在位置，从根节点出发，往左右儿子中找到合适位置。

删除：需要删除的节点没有左儿子，那么就把右儿子提上去。

        需要删除的节点的左儿子没有右儿子，那么就把左儿子提上去

　　　以上两种情况都不符合的话，就把左儿子的子孙中最大的节点提到需要删除的节点上。

#include <cstdio>

struct node {   //树  
    int val;  //节点的值
    node *lch, *rch;  //左右儿子
};

node *insert(node *p,int x) {
    if(p == NULL) {  //如果节点为空  并赋值为x
        node *q = new node; 
        q->val = x;
        q->lch = q->rch = NULL;
        return q;
    }
    else {   //递归调用
        if(x < p->val) p->lch = insert(p->lch,x);
        else p->rch = insert(p->rch,x);
        return p;
    }
}

bool find(node *p, int x) {
    if(p==NULL) return false;
    else if(x == p->val) return true;
    else if(x < p->val) return find(p->lch,x);
    else return find(p->rch,x);
}

node *remove(node *p,int x) {
    if(p==NULL) return NULL;  //如果树为空 返回NULL
    else if(x < p->val) p->lch = remove(p->lch,x); //
    else if(x > p->val) p->rch = remove(p->rch,x);
    else if(p->lch==NULL) {
        node *q=p->rch;
        delete p;
        return q;
    }
    else if(p->lch->rch == NULL) {
        node *q = p->lch;
        q->rch = p->rch;
        delete p;
        return q;
    }
    else {
        node *q;
        for(q = p->lch; q->rch->rch !=NULL; q = q->rch);
        node *r = q->rch;
        q->rch=r->lch;
        r->lch=p->lch;
        r->rch=p->rch;
        delete p;
        return r;
    }
    return p;
}

int main()
{
    node *root =NULL;
    root=insert(root,1);
    root=insert(root,2);
    root=insert(root,3);
    root=insert(root,5);
    root=insert(root,7);

    root=remove(root,2);

    bool flag=find(root,2);
    bool flag1=find(root,7);
    printf("%d %d
",flag,flag1);
    return 0;
}

set正是使用二叉搜索树维护集合的容器。

#include <cstdio>
#include <set>
using namespace std;

int main() {

    set<int> s;

    s.insert(1);
    s.insert(3);
    s.insert(5);

    set<int>::iterator it;

    it=s.find(1);
    if(it == s.end()) puts("not found");
    else puts("found");

    it=s.find(2);
    if(it == s.end()) puts("not found");
    else puts("find");

    s.erase(3);
    if(s.count(3)!=0) puts("found");
    else puts("not found");

    for(it = s.begin(); it != s.end(); it++) {
        printf("%d
",*it);
    }
    return 0;
}

map 是维护则是维护键和键对应值的容器。

#include <cstdio>
#include <map>
#include <string>
#include <iostream>
using namespace std;

int main() {

    map<int, const char*>m;

    m.insert(make_pair(1,"ONE"));
    m.insert(make_pair(10,"THE"));

    m.insert(make_pair(44,"HYNU"));
    m[100]="hynuacm";

    map<int,const char*>::iterator it;

    it=m.find(1);
    puts(it->second);
    it=m.find(2);
    if(it == m.end()) puts("not found");
    else puts(it->second);

    puts(m[10]);

    m.erase(10);

    for(it = m.begin(); it != m.end(); it++) {
        printf("%d: %s
",it->first,it->second);
    }
    return 0;
}