PAT习题解：1022. Digital Library (30)

题目链接：http://www.patest.cn/contests/pat-a-practise/1022

这道题目我没想到什么好办法做，最后参考了闲云阁的文章，用map 和 set 做出来，算是顺便学习了一下 STL
里面 map 和 set 的用法。话说C++的输入输出真是麻烦啊，这题用Python做的话写起来会很简单吧。

#define _CRT_SECURE_NO_WARNINGS
#include <map>
#include <set>
#include <string>
#include <iostream>
#include <cstring>
#include <cstdio>
using namespace std;

map<string, set<string>> bookname;
map<string, set<string>> author;
map<string, set<string>> keywords;
map<string, set<string>> publisher;
map<string, set<string>> year;

void printSearchResults(map<string, set<string>> &m, string query)
{
if (m[query].size() > 0)
{
for (set<string>::iterator it = m[query].begin(); it != m[query].end(); it++)
{
cout << *it << endl;
}
}
else
{
cout << "Not Found" << endl;
}
}

int main()
{
freopen("d:\\in.txt", "r", stdin);

char c;
string sid;
string s;
int n;
cin >> n;
for (int i = 0; i < n; i++)
{
cin >> sid;
c = getchar();
getline(cin, s);
bookname[s].insert(sid);

getline(cin, s);
author[s].insert(sid);

char str[105];
cin.getline(str, 105);
char *p = strtok(str," ");
while (p)
{
s = p;
keywords[s].insert(sid);
p = strtok(NULL, " ");
}

getline(cin, s);
publisher[s].insert(sid);

getline(cin, s);
year[s].insert(sid);
}

cin >> n;
int item;

while (n--)
{
scanf("%d: ", &item);
getline(cin, s);
cout << item << ": " << s << endl;
switch (item)
{
case 1:
printSearchResults(bookname, s);
break;
case 2:
printSearchResults(author, s);
break;
case 3:
printSearchResults(keywords, s);
break;
case 4:
printSearchResults(publisher, s);
break;
case 5:
printSearchResults(year, s);
break;
}
}
return 0;
}