题意: 找前缀数量 裸模板
1 #include<cstdio> 2 #include<cstring> 3 #include<iostream> 4 #include<algorithm> 5 #include<cmath> 6 #include<queue> 7 using namespace std; 8 const int maxn=500010; 9 const double eps=1e-8; 10 const long long modn=1000; 11 struct tri{ 12 int count; 13 int next[26]; 14 bool exist; 15 }e[maxn]; 16 char a[20]={}; 17 int tot=0; 18 void doit(int x,int k,int j){ 19 e[x].count++; 20 if(k<j){ 21 e[x].exist=1; 22 return; 23 } 24 int w=a[j]-'a'; 25 if(!e[x].next[w]){ 26 e[x].next[w]=++tot; 27 doit(tot,k,j+1); 28 }else{ 29 doit(e[x].next[w],k,j+1); 30 } 31 } 32 int getit(int x,int k,int j){ 33 if(k<j){ 34 return e[x].count; 35 } 36 int w=a[j]-'a'; 37 if(!e[x].next[w]){ 38 return 0; 39 } 40 else{ 41 return getit(e[x].next[w],k,j+1); 42 } 43 } 44 int main(){ 45 memset(e,0,sizeof(e)); 46 a[0]=getchar(); 47 while(1){ 48 int i=1;a[i]=getchar(); 49 while(a[i]!=' '){ 50 i++; 51 a[i]=getchar(); 52 } 53 doit(0,i-1,0); 54 a[0]=getchar(); 55 if(a[0]==' '){ 56 break; 57 } 58 } 59 while(~scanf("%s",&a)){ 60 int z=strlen(a)-1; 61 printf("%d ",getit(0,z,0)); 62 } 63 return 0; 64 }