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 + 10;
int trie[maxn][26], tot = 1, ed[maxn];
char s[maxn];

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

int searchh(char* str)
{
    int ans = 0;
    int len = strlen(str), p = 1;
    for(int k = 0; k < len; k++){
        
        p = trie[p][str[k] - 'a'];
        if(p == 0)return ans;
        ans += ed[p];
    }
    //ans += ed[p];
    return ans;
}

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

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