Trie樹（字典樹）

參考博客

1、引入

字典是幹啥的？查找字的。

字典樹天然也是起查找做用的。查找的是啥？單詞。

看如下幾個題：

一、給出n個單詞和m個詢問，每次詢問一個單詞，回答這個單詞是否在單詞表中出現過。

答：簡單！map，短小精悍。

好。下一個

二、給出n個單詞和m個詢問，每次詢問一個前綴，回答詢問是多少個單詞的前綴。

答：map，把每一個單詞拆開。

judge：n<=200000，TLE！

這就須要一種高級數據結構——Trie樹（字典樹）

2、原理

在本篇文章中，假設全部單詞都只由小寫字母構成

對cat，cash，app，apple，aply，ok 建一顆字典樹，建成以後以下圖所示

 

由此能夠看出：

一、字典樹用邊表示字母

二、有相同前綴的單詞公用前綴節點，那咱們能夠的得出每一個節點最多有26個子節點（在單詞只包含小寫字母的狀況下）

三、整棵樹的根節點是空的。爲何呢？便於插入和查找，這將會在後面解釋。

四、每一個單詞結束的時候用一個特殊字符表示，圖中用的‘′，那麼從根節點到任意一個‘′，那麼從根節點到任意一個‘’所通過的邊的全部字母表示一個單詞。

3、基本操做

A、insert，插入一個單詞

1.思路

  從圖中能夠直觀看出，從左到右掃這個單詞，若是字母在相應根節點下沒有出現過，就插入這個字母；不然沿着字典樹往下走，看單詞的下一個字母。

  這就產生一個問題：往哪兒插？計算機不會本身選擇位置插，咱們須要給它指定一個位置，那就須要給每一個字母編號。

  咱們設數組trie[i][j]=k，表示編號爲i的節點的第j個孩子是編號爲k的節點。

 什麼意思呢？

 這裏有2種編號，一種是i，k表示節點的位置編號，這是相對整棵樹而言的；另外一種是j，表示節點i的第j的孩子，這是相對節點i而言的。

 不理解？看圖

 仍是單詞cat，cash，app，apple，aply，ok 

 咱們就按輸入順序對其編第一種號，紅色表示編號結果。由於先輸入的cat，因此c，a，t分別是1,2,3，而後輸入的是cash，由於c，a是公共前綴，因此從s開始編，s是4，以此類推。

注意這裏相同字母的編號可能不一樣

第二種編號，相對節點的編號，紫色表示編號結果。

由於每一個節點最多有26個子節點，咱們能夠按他們的字典序從0——25編號，也就是他們的ASCLL碼-a的ASCLL碼。

注意這裏相同字母的編號相同

 實際上每一個節點的子節點都應該從0編到——25，但這樣會發現許多事根本用不到的。好比上圖的根節點應該分出26個叉。節約空間，用到哪一個分哪一個。

 這樣編號有什麼用呢？

回到數組trie[i][j]=k。 數組trie[i][j]=k，表示編號爲i的節點的第j個孩子是編號爲k的節點。

那麼第二種編號即爲j，第一種編號即爲i，k

二、代碼

 1 inline void insert(){
 2     len=strlen(s);
 3     root=0;
 4     for(int i=0;i<len;i++){
 5         int id=s[i]-'a';
 6         if(!trie[root][id])
 7                     trie[root][id]=++tot;
 8         root=trie[root][id];
 9     }
10 }

B、search，查找

查找有不少種，能夠查找某一個前綴，也能夠查找整個單詞。

再次咱們以查找一個前綴是否出現過爲例講解

一、思路

  從左往右以此掃描每一個字母，順着字典樹往下找，能找到這個字母，往下走，不然結束查找，即沒有這個前綴；前綴掃完了，表示有這個前綴。

二、代碼

 1 bool find(){
 2     len=strlen(s);
 3     root=0;
 4     for(int i=0;s[i];i++){
 5         int x=s[i]-'a';
 6         if(trie[root][x]==0)return false;
 7         root=trie[root][x];
 8     }
 9     return true;
10 }

三、若是是查詢某個單詞的話，咱們用bool變量 v[i]表示節點i是不是單詞結束的標誌。

    那麼最後return的是v[root],因此在插入操做中插入完每一個單詞是，要對單詞最後一個字母的v[i]置爲true，其餘的都是false

四、若是是查詢前綴出現的次數的話，那就在開一個sum[]，表示位置i被訪問過的次數，

   那麼最後return的是sum[root]，插入操做中每訪問一個節點，都要讓他的sum++

   這裏前綴的次數是標記在前綴的最後一個字母所在位置的後一個位置上。

  好比:前綴abc出現的次數標記在c所在位置的後一個位置上，

4、完整代碼

一、查詢是否出現

 1 #include<bits/stdc++.h>
 2 #define maxn 2000010
 3 using namespace std;
 4 int tot=1,n;
 5 int trie[maxn][26];
 6 //bool isw[maxn];查詢整個單詞用
 7 void insert(char *s,int rt){
 8     for(int i=0;s[i];i++){
 9         int x=s[i]-'a';
10         if(trie[rt][x]==0){
11             trie[rt][x]=++tot;
12         }
13         rt=trie[rt][x]; 
14     }
15 }
16 bool find(char *s,int rt){
17     for(int i=0;s[i];i++){
18         int x=s[i]-'a';
19         if(trie[rt][x]==0)return false;
20         rt=trie[rt][x];
21     }
22     return true;
23 }
24 char s[22];
25 int main(){
26     tot=0;
27     int rt=1;
28     scanf("%d",&n);
29     for(int i=1;i<=n;i++){
30         cin>>s;
31         insert(s,rt);
32     }
33     scanf("%d",&n);
34     for(int i=1;i<=n;i++){
35         cin>>s;
36         if(find(s,rt))printf("YES\n");
37         else printf("NO\n");
38     }
39     return 0;
40 }

二、查詢前綴出現次數

 1 #include<bits/stdc++.h>
 2 using namespace std;
 3 int trie[400001][26],len,root,tot,sum[400001];
 4 bool p;
 5 int n,m; 
 6 char s[11];
 7 void insert(){
 8     len=strlen(s);
 9     root=0;
10     for(int i=0;i<len;i++){
11         int id=s[i]-'a';
12         if(!trie[root][id]) trie[root][id]=++tot;
13         sum[trie[root][id]]++;
14         root=trie[root][id];
15     }
16 }
17 int search(){
18     root=0;
19     len=strlen(s);
20     for(int i=0;i<len;i++){
21         int id=s[i]-'a';
22         if(!trie[root][id]) return 0;
23         root=trie[root][id];
24     }
25     return sum[root];
26 }
27 int main(){
28     scanf("%d",&n);
29     for(int i=1;i<=n;i++){
30         cin>>s;
31         insert();
32     }
33     scanf("%d",&m);
34     for(int i=1;i<=m;i++){
35         cin>s;
36         printf("%d\n",search());
37     }
38 }

完整代碼

 1 #include<bits/stdc++.h>
 2 using namespace std;
 3 char s[11];
 4 int n,m;
 5 bool p;
 6 struct node{
 7     int count;
 8     node * next[26];
 9 }*root;
10 node * build(){
11     node * k=new(node);
12     k->count=0;
13     memset(k->next,0,sizeof(k->next));
14     return k;
15 }
16 void insert(){
17     node * r=root;
18     char * word=s;
19      while(*word){
20         int id=*word-'a';
21         if(r->next[id]==NULL) r->next[id]=build();
22         r=r->next[id];
23         r->count++;
24         word++;
25     }
26 }
27 int search(){
28     node * r=root;
29     char * word=s;
30     while(*word){
31         int id=*word-'a';
32         r=r->next[id];
33         if(r==NULL) return 0;
34         word++;
35     }
36     return r->count;
37 }
38 int main(){
39     root=build();
40     scanf("%d",&n);
41     for(int i=1;i<=n;i++){
42             cin>>s;
43             insert();
44     }
45     scanf("%d",&m);
46     for(int i=1;i<=m;i++){
47         cin>>s;
48         printf("%d\n",search());
49     }
50 }