家装e站杭州小程序建设

原题

1022 Digital Library - PAT (Advanced Level) Practice

题目大意

给定一本书的id、书名、作者、(多个)关键词、出版社、出版年份，根据输入的书名、作者、(其中一个)关键词、出版社、出版年份按大小顺序返回id

解题思路

先吐槽一句，题目简单是简单，麻烦是真麻烦...

利用哈希表来储存元素，由于可能多本书有相同的关键字，因此用vector<string>来储存答案的id。

另外，对于keywords，我们要额外进行空格分割，再分别将每一个关键词存入哈希表。

代码(c++)

#include <bits/stdc++.h>
#include <map>
#include <vector>
#include <algorithm>using namespace std;const int N = 10010;int n, m;
unordered_map<string, vector<string>> title_m;        // 书名哈希表
unordered_map<string, vector<string>> author_m;       // 作者哈希表
unordered_map<string, vector<string>> key_m;          // 关键词哈希表
unordered_map<string, vector<string>> publisher_m;    // 出版社哈希表
unordered_map<string, vector<string>> year_m;         // 年份哈希表void split(string s, unordered_map<string, vector<string>> &m, string id) {  // 根据空格分割关键词int len = 0;for(int i = 0; i < s.size(); i++) {if(s[i] == ' ') {string str = s.substr(i - len, len);m[str].push_back(id);len = -1;                            // len为-1因为马上下面就要+1了}len++;}string str = s.substr(s.size() - len);m[str].push_back(id);
}bool find_id(const string& search, const unordered_map<string, vector<string>>& m) {auto it = m.find(search); if (it != m.end())                            // 找到，排序输出并返回truevector<string> res = it->second;  sort(res.begin(), res.end());for (string id : res) cout << id << endl;return true;                    }return false;
}int main() {string sn;getline(cin, sn);                                    // 此处用getline读入，防止有其他空格n = atoi(sn.c_str());for(int i = 0; i < n; i++) {string id, title, author, key, publisher, year;getline(cin, id);getline(cin, title);getline(cin, author);getline(cin, key);getline(cin, publisher);getline(cin, year);title_m[title].push_back(id);author_m[author].push_back(id);publisher_m[publisher].push_back(id);split(key, key_m, id);year_m[year].push_back(id);}string sm;getline(cin, sm);m = atoi(sm.c_str());vector<string> ans;for (int i = 0; i < m; i++) {string search_id;getline(cin, search_id);cout << search_id << endl;string search = search_id.substr(3, search_id.size() - 3);  // 注意输入的search是带id的，先去掉if (find_id(search, title_m)) continue;if (find_id(search, author_m)) continue;if (find_id(search, key_m)) continue;if (find_id(search, publisher_m)) continue;if (find_id(search, year_m)) continue;cout << "Not Found" << endl;}
}