【算法笔记】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 = 0; 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 = 0; i < m; i++){
        scanf("%d: ", &type);
        getline(cin, search);
        cout<<type<<": "<<search<<endl;
        if(type == 1) query(Title, search);
        else if(type == 2) query(Author, search);
        else if(type == 3) query(Word, search);
        else if(type == 4) query(Publishier, search);
        else if(type == 5) query(Year, search);
    }
    return 0;
}