字符串-trie模板题

点击打开链接

非常好的模板题

点击打开链接

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
using namespace std;
int trie[400001][26],len,root,tot,sum[400001];
bool p;
int n,m;
char s[11];
void insert(char *s)
{
    len=strlen(s);
    root=0;
    for(int i=0;i<len;i++)
    {
        int id=s[i]-'a';
        if(!trie[root][id]) trie[root][id]=++tot;
        sum[trie[root][id]]++;//前缀后移一个位置保存
        root=trie[root][id];
    }
}
int search(char *s)
{
    root=0;
    len=strlen(s);
    for(int i=0;i<len;i++)
    {
        int id=s[i]-'a';
        if(!trie[root][id]) return 0;
        root=trie[root][id];
    }//root经过此循环后变成前缀最后一个字母所在位置的后一个位置
    return sum[root];//因为前缀后移了一个保存，所以此时的sum[root]就是要求的前缀出现的次数
}
int main(){

    while(gets(s)){
        if(strcmp(s,"") == 0)
            break;
       insert(s);
    }
    while(scanf("%s",s) != EOF){
        cout<<search(s)<<endl;
    }

    return 0;
}