PAT甲级1022 Digital Library

题目：https://pintia.cn/problem-sets/994805342720868352/problems/994805480801550336

题意：

每一本书有一个id， 书名，作者，至多五个关键字，一个出版社名，出版社年份。

现在根据给定的书名或作者或关键字或出版社名或年份，按照id字典序大小输出符合条件的书。

思路：

对每一个属性都用map维护。

一个坑点是，前面说书的年份一定在1000-3000之间，但是查询的时候的年份不一定满足，而且这里输出的时候也要满足4位。测试点1就是和年份有关的。

 #include<cstdio>
 #include<cstdlib>
 #include<map>
 #include<set>
 #include<iostream>
 #include<cstring>
 #include<algorithm>
 #include<vector>
 #include<cmath>
 #include<stack>
 #include<queue>

 #define inf 0x7fffffff
 using namespace std;
 typedef long long LL;
 typedef pair<string, string> pr;

 int n, m;
 const int maxn = 1e4 + ;

 struct node{
     string id;
     string title;
     string author;
     string keyword[];
     string publisher;
     int year;
     int keynum = ;
 }book[maxn];

 map<string, int>titlemp;
 map<string, int>authormp;
 map<string, int>keymp;
 map<string, int>publishermp;
 map<int, string>idmp;
 int titletot, authortot, keytot, publishertot, idtot;

 vector<int> title[maxn], author[maxn], keyword[], publisher[], year[];

 bool cmp(int a, int b)
 {
     int ida = stoi(idmp[a]), idb = stoi(idmp[b]);
     return ida < idb;
 }

 int main()
 {
     scanf("%d", &n);
     for(int i = ; i < n; i++){
         getchar();getline(cin, book[i].id);
         idmp[i] = book[i].id;

         getline(cin, book[i].title);
         if(titlemp.find(book[i].title) == titlemp.end()){
             titlemp[book[i].title] = titletot++;
         }
         title[titlemp[book[i].title]].push_back(i);
         getline(cin, book[i].author);
         if(authormp.find(book[i].author) == authormp.end()){
             authormp[book[i].author] = authortot++;
         }
         author[authormp[book[i].author]].push_back(i);
         char ch;
         while(){
             cin>>book[i].keyword[book[i].keynum];
             //cout<<book[i].keyword[book[i].keynum]<<endl;
             if(keymp.find(book[i].keyword[book[i].keynum]) == keymp.end()){
                 keymp[book[i].keyword[book[i].keynum]] = keytot++;
             }
             keyword[keymp[book[i].keyword[book[i].keynum++]]].push_back(i);
             ch = getchar();
             if(ch == '\n')break;
         }
         getline(cin, book[i].publisher);
         if(publishermp.find(book[i].publisher) == publishermp.end()){
             publishermp[book[i].publisher] = publishertot++;
         }
         publisher[publishermp[book[i].publisher]].push_back(i);
         cin>>book[i].year;
         year[book[i].year - ].push_back(i);
     }

     scanf("%d", &m);
     while(m--){
         int type;
         scanf("%d: ", &type);
         string s;
         int id, y;
         //getchar();
         switch(type){
             case :
                 getline(cin, s);
                 printf("1: ");cout<<s<<endl;
                 if(titlemp.find(s) == titlemp.end()){
                     printf("Not Found\n");
                 }
                 else{
                     id = titlemp[s];
                     sort(title[id].begin(), title[id].end(), cmp);
                     for(int i = ; i < title[id].size(); i++){
                         cout<<idmp[title[id][i]]<<endl;
                     }
                 }
                 break;
             case :
                 getline(cin, s);
                 printf("2: ");cout<<s<<endl;
                 if(authormp.find(s) == authormp.end()){
                     printf("Not Found\n");
                 }
                 else{
                     id = authormp[s];
                     sort(author[id].begin(), author[id].end(), cmp);
                     for(int i = ; i < author[id].size(); i++){
                         cout<<idmp[author[id][i]]<<endl;
                     }
                 }
                 break;
             case :
                 getline(cin, s);
                 printf("3: ");cout<<s<<endl;
                 if(keymp.find(s) == keymp.end()){
                     printf("Not Found\n");
                 }
                 //cout<<id<<endl;
                 else{
                     id = keymp[s];
                     sort(keyword[id].begin(), keyword[id].end(), cmp);
                     for(int i = ; i < keyword[id].size(); i++){
                         cout<<idmp[keyword[id][i]]<<endl;
                     }
                 }
                 break;
             case :
                 getline(cin, s);
                 printf("4: ");cout<<s<<endl;
                 if(publishermp.find(s) == publishermp.end()){
                     printf("Not Found\n");
                 }
                 else{
                     id = publishermp[s];
                     sort(publisher[id].begin(), publisher[id].end(), cmp);
                     for(int i = ; i < publisher[id].size(); i++){
                         cout<<idmp[publisher[id][i]]<<endl;
                     }
                 }
                 break;
             case :
                 cin>>s;
                 printf("5: ");cout<<s<<endl;
                 y = stoi(s);
                 if( y< || y > || year[y - ].size() ==  ){
                     printf("Not Found\n");
                 }
                 else{
                     sort(year[y - ].begin(), year[y - ].end(), cmp);
                     for(int i = ; i < year[y - ].size(); i++){
                         cout<<idmp[year[y - ][i]]<<endl;
                     }
                 }
                 break;
         }
     }
     return ;
 }