CH601后缀数组【Trie树】

内含字典树创建及查询模板

1601 前缀统计 0x10「基本数据结构」例题

描述

给定N个字符串S1,S2...SN，接下来进行M次询问，每次询问给定一个字符串T，求S1～SN中有多少个字符串是T的前缀。输入字符串的总长度不超过10^6，仅包含小写字母。

输入格式

第一行两个整数N，M。接下来N行每行一个字符串Si。接下来M行每行一个字符串表示询问。

输出格式

对于每个询问，输出一个整数表示答案

样例输入

3 2
ab
bc
abc
abc
efg

样例输出

2
0

思路：

用scanf ！=EOF最后一组不知道为什么就是会RE

虐狗宝典字典树笔记：

Trie树，字典树每个节点拥有若干个字符指针，在插入或检索字符串时扫到一个字符c就沿着当前节点的c字符指针走下去。

初始时，空Trie仅包含一个根节点，字符指针均为空。

插入S时，我们令一个指针p指向根节点，依次扫描S中的每一个字符c。

　　1.若p的c字符指针指向一个已经存在的节点Q，则令P=Q

　　2.若p的c字符指针指向空，则新建一个节点Q，令p的c字符指针指向Q，然后令P=Q

　　当S扫描完毕，当前节点P上标记他是一个字符串的末尾

检索S是否在Trie中存在，令一个指针P指向根节点，依次扫描S中的每个字符c

　　1.若P的c字符指针指向空，则说明S没有被插入过Trie，结束检索

　　2.若P的c字符指针指向一个已经存在的节点Q,则令P=Q

　　当S中的字符扫描完毕，若当前节点P被标记为一个字符串的末尾，说明S在Trie中存在。

本题与讲解的不同之处在于末尾应标记出现的次数，检索时也应该同时统计所经过节点的作为末尾的次数。

 #include <iostream>
 #include <set>
 #include <cmath>
 #include <stdio.h>
 #include <cstring>
 #include <algorithm>
 #include <map>
 using namespace std;
 typedef long long LL;
 #define inf 0x7f7f7f7f

 int n, m;
 const int maxn = 1e6 + ;
 int trie[maxn][], tot = , ed[maxn];
 char s[maxn];

 void insertt(char *str)
 {
     int len = strlen(str), p = ;
     for(int k = ; k < len; k++){
         int ch = str[k] - 'a';
         if(trie[p][ch] == ){
             trie[p][ch] = ++tot;
         }
         p = trie[p][ch];
     }
     ed[p]++;
 }

 int searchh(char* str)
 {
     int ans = ;
     int len = strlen(str), p = ;
     for(int k = ; k < len; k++){

         p = trie[p][str[k] - 'a'];
         if(p == )return ans;
         ans += ed[p];
     }
     //ans += ed[p];
     return ans;
 }

 int main()
 {
     scanf("%d%d", &n, &m);

     for(int i = ; i <= n; i++){
         scanf("%s", s);
         insertt(s);
     }
     for(int j = ; j <= m; j++){
         scanf("%s", s);
         printf("%d\n", searchh(s));
     }
     return ;
 }