【算法笔记】A1022 Digital Library

题意

　　输入n本书的信息：id，书名，作者，关键字，出版社，出版年份。搜索图书，输出id。

思路

　　定义5个map<string, set<int> >，分别存放Title, Author, Word, Publishier, Year与id的映射关系，然后只需要考虑怎么输入就可以了。注意因为字符串和map的参数传递很慢，所以如果把查询写成函数，必须对参数进行引用，否则会导致运行超时。

code：

 #include<bits/stdc++.h>
 using namespace std;
 map<string, set<int> > Title, Author, Word, Publishier, Year;
 void query(map<string, set<int> > &mp, string &str){
     if(mp.find(str) == mp.end()) cout<<"Not Found"<<endl;
     else{
         for(set<int>::iterator it = mp[str].begin(); it!=mp[str].end(); it++){
             printf("%07d\n", *it);
         }
     }
 }
 int main(){
     int n, m, id, type;
     string tit, aut, word, pub, year, search;
     cin>>n;
     for(int i = ; i < n; i++){
         cin>>id;
         getchar();
         getline(cin, tit);
         Title[tit].insert(id);
         getline(cin, aut);
         Author[aut].insert(id);
         while(cin>>word){
             Word[word].insert(id);
             char c = getchar();
             if(c == '\n') break;
         }
         getline(cin, pub);
         Publishier[pub].insert(id);
         getline(cin, year);
         Year[year].insert(id);
     }
     cin>>m;
     for(int i = ; i < m; i++){
         scanf("%d: ", &type);
         getline(cin, search);
         cout<<type<<": "<<search<<endl;
         if(type == ) query(Title, search);
         else if(type == ) query(Author, search);
         else if(type == ) query(Word, search);
         else if(type == ) query(Publishier, search);
         else if(type == ) query(Year, search);
     }
     return ;
 }