HDU 1251统计难题

 1 #include<iostream>
 2 #include<cstdio>
 3 #include<cstring>
 4 #include<vector>
 5 using namespace std;
 6 vector<string>vec[26];
 7 int cnt,flag=0;int len[26];
 8 string str;
 9 int main()
10 {
11     while(getline(cin,str)){
12         if(str==""){
13             flag=1;
14             for(int i=0;i<26;i++)len[i]=vec[i].size();
15             continue;
16         }
17         if(!flag)  vec[str[0]-‘a‘].push_back(str);
18         else if(flag){
19             cnt=0;
20             int tmp=str[0]-‘a‘;
21             for(int i=0;i<len[tmp];i++){
22                 int len2=str.length();int j;
23                 for(j=1;j<len2;j++)
24                   if(str[j]!=vec[tmp][i][j])break;
25                   if(j==len2) cnt++;
26             }
27             printf("%d\n",cnt);  
28         }
29     }
30     return 0;
31 }