字典树

 1 #include <iostream>
 2 #include <string>
 3 using namespace std;
 4 struct Node
 5 {
 6     Node *ap[26];//保存指向下一节点的指针
 7     //---
 8     //存放每个单词的一些属性，数据类型根据需求自定，或者使用结构体保存
 9     int num;//保存出现次数
10     string val;//保存当前单词对应的内容
11     //---
12     void ini()//生成新节点时调用对其初始化
13     {
14         num=0;
15         for(int i=0;i<26;i++)
16             ap[i]=NULL;
17     }
18 }*head;//申明头指针用于指向字典树
19 void Insert(string str,string val)
20 {
21     Node *now=head,*makeNode;//now指针用于指向当前位置，makeNode用于生成新节点，再让now->ap[odx]指向新生成的节点
22     for(int i=0;i<str.length();i++)
23     {
24         int odx=str[i]-‘a‘;
25         if(now->ap[odx]==NULL)//如果为空说明字典树没有后续的内容，就建立新节点
26         {
27             makeNode=new Node;
28             makeNode->ini();
29             now->ap[odx]=makeNode;
30         }
31         now=now->ap[odx];//迭代now指针，使其转移到下一位置
32         now->num++;
33         if(i==str.length()-1)//当i满足条件时，此时位置为插入单词的末尾，故可以用当前节点代表这个单词，把要存储的对应信息存入此节点
34             now->val=val;
35     }
36 }
37 string Find(string str)//返回值根据要求而定
38 {
39     int counter=0;
40     Node *now=head;
41     for(int i=0;i<str.length();i++)//迭代过程类似插入
42     {
43         int odx=str[i]-‘a‘;
44         counter=now->num;
45         if(now->ap[odx]==NULL)
46             break;
47         now=now->ap[odx];
48         if(i==str.length()-1)
49             return now->val;
50     }
51     return "";
52 }
53 void getstr(string &temp,string &str,string &val)
54 {
55     unsigned long len=temp.length();
56     int pos;
57     for(int i=0;i<len;i++)
58     {
59         if(temp[i]!=‘ ‘)
60             str+=temp[i];
61         else
62         {
63             pos=i+1;
64             break;
65         }
66     }
67     for(int i=pos;i<len;i++)
68         val+=temp[i];
69 }
70 int main(int argc, const char * argv[]) {
71     string str,val,temp;
72     head=new Node;
73     head->ini();
74     while(getline(cin,temp))
75     {
76         if(temp.length()==0)
77             break;
78         str=val="";
79         getstr(temp,str,val);
80         Insert(val,str);
81     }
82     while(cin>>str)
83     {
84         temp=Find(str);
85         if(temp.length()==0)
86             cout<<"eh"<<endl;
87         else
88             cout<<Find(str)<<endl;
89     }
90     return 0;
91 }