hdu1251（字典树模板）

版权声明：本文为博主原创文章，转载请说明出处。 https://blog.csdn.net/xianpingping/article/details/83096900

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const int maxn=2000000+10;
const int inf=0x3f3f3f3f;
int tree[maxn][30];
bool flagg[maxn];
int sum[maxn];///有时要加这个数组，目的是为了统计以该节点为前缀的单词的数量
int tot;
void insert_(string str){
    int len=str.size();
    int root=0;
    for(int i=0;i<len;i++){
        int id=str[i]-'a';
        if(!tree[root][id]){
            tree[root][id]=++tot;
        }
        root=tree[root][id];
        sum[root]++;
    }
    flagg[root]=true;
}
int find_(string str){
    int len=str.size();
    int root=0;
    for(int i=0;i<len;i++){
        int id=str[i]-'a';
        if(!tree[root][id])
            return 0;///该单词没有出现过
        root=tree[root][id];
    }
    /*if(flagg[root])
        return sum[root];///出现过，不再算贡献
    else
        return 0;///没有出现过
        */
    return sum[root];
}
char ss[maxn];
void init(){
    memset(sum,0,sizeof(sum));
    tot=0;
   /// memset(flagg,0,flagg);
}
int main()
{
    init();
    int cnt=0;
    while(gets(ss)){
        if(ss[0]=='\0')
            break;
        insert_(ss);
    }
    ///输入空行后。
    while(scanf("%s",ss)!=EOF){
        printf("%d\n",find_(ss));
    }
    return 0;
}