HDU1251 统计难题 trie树 简单

http://acm.hdu.edu.cn/showproblem.php?pid=1251

题意: 找前缀数量 裸模板

 

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<cmath>
#include<queue>
using namespace std;
const int maxn=500010;
const double eps=1e-8;
const long long modn=1000;
struct tri{
    int count;
    int next[26];
    bool exist;
}e[maxn];
char a[20]={};
int tot=0;
void doit(int x,int k,int j){
    e[x].count++;
    if(k<j){
        e[x].exist=1;
        return;
    }
    int w=a[j]-'a';
    if(!e[x].next[w]){
        e[x].next[w]=++tot;
        doit(tot,k,j+1);
    }else{
        doit(e[x].next[w],k,j+1);
    }
}
int getit(int x,int k,int j){
    if(k<j){
        return e[x].count;
    }
    int w=a[j]-'a';
    if(!e[x].next[w]){
        return 0;
    }
    else{
        return getit(e[x].next[w],k,j+1);
    }
}
int main(){
    memset(e,0,sizeof(e));
    a[0]=getchar();
    while(1){
        int i=1;a[i]=getchar();
        while(a[i]!='\n'){
            i++;
            a[i]=getchar();
        }
        doit(0,i-1,0);
        a[0]=getchar();
        if(a[0]=='\n'){
            break;
        }
    }
    while(~scanf("%s",&a)){
        int z=strlen(a)-1;
        printf("%d\n",getit(0,z,0));
    }
    return 0;
}