字典树的操作

#include <iostream>
#include <cstdlib>
#include <string>
using namespace std;
#define MAX 26
typedef struct _Trie
{
int num;
struct _Trie *next[MAX];
}Trie,*PTrie;
PTrie root;
void InsertTrie(string str)
{
int len=str.length();
int i,j,pos;
PTrie p=root;
PTrie q;
for(i=0;i<len;i++)
{
pos=str[i]-'a';
if(p->next[pos]==NULL)
{
q=(PTrie)malloc(sizeof(Trie));
q->num=1;
for(int j=0;j<MAX;j++)
{
q->next[j]=NULL;
}
p->next[pos]=q;
p=p->next[pos];
}
else
{
p->next[pos]->num++;
p=p->next[pos];

}

}
p->num=0;
}
//
int FindTrie(string str)
{
int i,pos;
int len=str.length();
PTrie p=root,q;

for(i=0;i<len;i++)
{
pos=str[i]-'a';
if(p->next[pos]==NULL)
{
return -1;//代表没有这样的前缀
}
else
{
p=p->next[pos];
}
}
return p->num;
}
int main()
{
root=(PTrie)malloc(sizeof(Trie));
root->num=1;
for(int i=0;i<MAX;i++)
{
root->next[i]=NULL;
}
string str1="abc";
string str2="abcd";
string str3="abd";
string str4="b";
string str5="bcd";
string str6="efg";
string hig="hig";
InsertTrie(str1);
InsertTrie(str2);
InsertTrie(str3);
InsertTrie(str4);
InsertTrie(str5);
InsertTrie(str6);
if(FindTrie("ab")==-1)
cout<<"没有以这个为前缀的字符串"<<endl;
else
cout<<"以这个子串为前缀的字符串个数为"<<FindTrie("ab")<<endl;

return 0;
}