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 + 5;

struct node{
    string id;
    string title;
    string author;
    string keyword[5];
    string publisher;
    int year;
    int keynum = 0;
}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[1005], publisher[1005], year[2500];

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