字典树trie_tree
闲着没事，把以前写的字典树搬过来，贴一下。
深搜遍历那块个人觉得没有写太好，因为其余都是用char *实现的，那部分却用了string.
char*的速度是比string快很多的，所以还有待改进。所有代码均属个人编写，没有经过专门
测试，所以暂时还不知道正确性，如果有问题，还请指正。
#include<iostream>
#include<string>
using namespace std;
struct trie_node
{
trie_node()
{
is_word = false;
for(int i = 0;i<26;i++)
{
node[i] = NULL;
}
}
bool is_word ; // 根到该节点是否对应一个字符串
trie_node *node[26];
};
class Trie
{
public:
trie_node *root; //根节点
Trie(){root = NULL;}
bool trie_search(const char *str) //查找
{
trie_node *temp = root;
for(int i = 0;i<strlen(str);i++)
{
if(temp->node[str[i]-'a'] == NULL)
return false;
temp = temp->node[str[i]-'a'];
}
return temp->is_word;
};
void trie_insert(const char *str) //插入
{
if(root == NULL)
root = new trie_node;
trie_node *temp = root;
for(int i = 0;i<strlen(str);i++)
{
int site = str[i] - 'a';
if(temp->node[site] == NULL)
{
temp->node[site] = new trie_node;
temp = temp->node[site];
}
else
temp = temp->node[site];
}
temp->is_word = true;
};
bool trie_del(const char *str) //删除
{
trie_node *temp = root;
for(int i = 0;i<strlen(str);i++)
{
if(temp->node[str[i]-'a'] == NULL)
return false;
temp = temp->node[str[i]-'a'];
}
if(temp->is_word)
temp->is_word = false;
else
return false;
return true;
}
};
void print(trie_node *root,string str)
{
for(int i = 0;i<26;i++)
{
if(root->node[i] != NULL)
{
trie_node *temp = root;
root = root->node[i];
str += i+'a';
if(root->is_word)
{
cout<<str<<endl;
}
print(root,str);
str.erase(str.size()-1,1); //删除最后一个元素
root = temp;
}
}
}
int main()
{
char s[11];
int n;
string str2 ;
printf("1.插入/n");
printf("2.查找/n");
printf("3.删除/n");
printf("4.深度遍历/n");
Trie x;
while(scanf("%d",&n)!=EOF)
{
switch(n)
{
case 1:
scanf("%s",&s);
x.trie_insert(s);
printf("insert succeed/n");
break;
case 2:
scanf("%s",&s);
if(x.trie_search(s))
printf("yes/n");
else
printf("no/n");
break;
case 3:
scanf("%s",&s);
if(x.trie_del(s))
printf("yes/n");
else
printf("no/n");
break;
case 4:
str2 = "";
print(x.root,str2);
break;
default:
;
}
printf("1.插入/n");
printf("2.查找/n");
printf("3.删除/n");
printf("4.深度遍历/n");
}
return 0;
}