Trie树

Posted 2021-01-06 mr94kevin

一看名字就知道这是棵树，而且和字符串有关。

就我理解，Trie树是以字符为边，以字符串信息为点的树。可以用于字符串的去重、检索等。

实现的话，将根节点认为是一个特殊的点，从此开始向下保存字符串。设ch[i][j]存储结点i的字符编号为j的孩子的编号，cnt[i]表示以结点i作为结束的字符串数（也可以是其他用途），num为树的结点总数。

这样可以对应每个字符串将其保存在树里，自动去重。查找时，只需对应着边往下找即可。

 1 struct Trie {
 2     int ch[maxn][26],cnt[maxn],num;
 3     Trie() {
 4         memset(ch[0],0,sizeof(ch[0]));
 5         memset(cnt,0,sizeof(cnt));
 6         num=0;
 7     }
 8     int newnode() {
 9         ++num;
10         memset(ch[num],0,sizeof(ch[num]));
11         return num;
12     }
13     void insert(char* s) {
14         int u=0;
15         for(int i=0;s[i];++i) {
16             if(!ch[u][s[i]-‘a‘])
17                 ch[u][s[i]-‘a‘]=newnode();
18             u=ch[u][s[i]-‘a‘];
19         }
20         ++cnt[u];
21     }
22     int query(char* s) {
23         int u=0;
24         for(int i=0;s[i];++i) {
25             if(!ch[u][s[i]-‘a‘]) return 0;
26             u=ch[u][s[i]-‘a‘];
27         }
28         return cnt[u];
29     }
30 };