PAT1022. Digital Library

#include <iostream>
#include<cstdio>
#include<map>
#include<string>
#include<set>
using namespace std;
int n;
map<string,set<int> >mptitle,mpauthor,keyword,publisher,year;


void cal(map<string,set<int> >&mp,string&str){
if(mp.find(str)==mp.end())cout<<"Not Found"<<endl;
else{
set<int>::iterator it=mp[str].begin();
for(;it!=mp[str].end();it++){
printf("%07d\n",*it);
}

}

}

int main(){
    //freopen("d://jin.txt","r",stdin);
string title,author,key,publish,y;
char c;
cin>>n;
int d;
for(int i=0;i<n;i++){
cin>>d;
c=getchar();
getline(cin,title);
mptitle[title].insert(d);
getline(cin,author);
mpauthor[author].insert(d);
while(cin>>key){
keyword[key].insert(d);
c=getchar();
if(c=='\n')break;}
getline(cin,publish);
publisher[publish].insert(d);
getline(cin,y);
year[y].insert(d);
}
string temp;
int m;
cin>>m;
int t;
for(int i=0;i<m;i++){
scanf("%d: ",&t);

getline(cin,temp);
cout<<t<<": "<<temp<<endl;
if(t==1){
cal(mptitle,temp);
}
else if(t==2){cal(mpauthor,temp);}
else if(t==3){cal(keyword,temp);}
else if(t==4){cal(publisher,temp);;}
else{cal(year,temp);;}


}





    return 0;
}