hdu 1251 统计难题

#include<bits/stdc++.h>
using namespace std;
map<string,int>pq;
int main()
{
string str,str1;
while(getline(cin,str),str!="")
{
for(int i=0;i<str.size();i++)
{
if(pq.count(str.substr(0,i+1))==0)
pq[str.substr(0,i+1)]=0;
pq[str.substr(0,i+1)]++;
}
}
int k=0;
while(cin>>str1)
{
cout<<pq[str1]<<"\n";
}
return 0;
}