统计难题 HDU - 1251

 

给一些字符串，问以某个串为前缀的串有几个

 

?

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45

#include<iostream> #include <algorithm> #include <cstring> using namespace std;  const int N=5e5+4;  char num[80];  int len,sum[N];  int ch[N][30],tot;     int find(char *s){     int i,u=1;           for(i=0;i<len;i++){          int c=s[i]-'a';          if(ch[u][c]==0){              return 0;          }          u=ch[u][c];      }      return sum[u];  }  void insert(char *s){      int i,u=1;             for(i=0;i<len;i++){          int c=s[i]-'a';          if(ch[u][c]==0){              ch[u][c]=++tot;          }          u=ch[u][c];          sum[u]++;      }  }     int main(){      tot=1;          memset(ch,0,sizeof ch);                  while(gets(num),strlen(num)>0){           len=strlen(num); insert(num);       }       while(gets(num)){           len=strlen(num); cout<<find(num)<<'\n';       }  }