PAT 1022. Digital Library

完全考输入输出

#include <iostream>

#include <cstdio>

#include <cstdlib>

#include <vector>

#include <string>

#include <unordered_map>

#include <algorithm>

using namespace std;



unordered_map<string, vector<int> > idx_title;

unordered_map<string, vector<int> > idx_author;

unordered_map<string, vector<int> > idx_keyword;

unordered_map<string, vector<int> > idx_publisher;

unordered_map<string, vector<int> > idx_year;



unordered_map<string, vector<int> >* idx[6] = {NULL, &idx_title, &idx_author, &idx_keyword, &idx_publisher, &idx_year};



void idx_add_book(unordered_map<string, vector<int> >& idx, const string& key, int book_id) {

    auto iter = idx.find(key);

    if (iter == idx.end()) {

        iter = idx.insert(make_pair(key, vector<int>())).first;

    }

    iter->second.push_back(book_id);

}



int main() {

    

    int N, M;

    scanf("%d", &N);



    char ibuf[100] = {0};

    int bid = 0;

    for (int i=0; i<N; i++) {

        scanf("%d", &bid);

        getchar();

        // read title

        scanf("%[^\n]", ibuf);

        idx_add_book(idx_title, string(ibuf), bid);

        getchar();

        // read author

        scanf("%[^\n]", ibuf);

        idx_add_book(idx_author, string(ibuf), bid);

        getchar();

        // read keywords

        for (;;) {

            scanf("%[^ \n]", ibuf);

            idx_add_book(idx_keyword, string(ibuf), bid);

            if (getchar() == '\n') {

                break;

            }

        }

        // read publisher

        scanf("%[^\n]", ibuf);

        idx_add_book(idx_publisher, string(ibuf), bid);

        getchar();

        // read year

        scanf("%[^\n]", ibuf);

        idx_add_book(idx_year, string(ibuf), bid);

        getchar();

    }



    scanf("%d", &M);

    string buf;

    for (int i=0; i<M; i++) {

        int idx_type = 0;

        scanf("%d", &idx_type);

        getchar();

        getchar();

        if (idx_type < 1 || idx_type > 5) {

            break;

        }

        getline(cin, buf);

        printf("%d: %s\n", idx_type, buf.c_str());

        auto dict = idx[idx_type];

        if (dict == NULL) {

            break;

        }

        auto iter = dict->find(buf);

        if (iter == dict->end()) {

            printf("Not Found\n");

            continue;

        }

        auto v = iter->second;

        sort(v.begin(), v.end());

        int len = v.size();

        for (int i=0; i<len; i++) {

            printf("%07d\n", v[i]);

        }

    }



    return 0;

}