【PAT甲级】1022. Digital Library (30)

#include <iostream>
#include <iomanip>
#include <sstream>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <string>
#include <map>
#include <set>
using namespace std;
int n;
int m;
map<string, set<int> > lib[6];

int main(int argc, char *argv[]) {
    int i, j, k;
    cin >> n;
    int count = 0;
    for (i = 0; i < n; i++) {
        int id;
        cin >> id;
        getchar();//把缓冲中的回车清除掉
        for (j = 1; j < 6; j++) {
            string line;
            if (j != 3) {
                getline(cin, line);
                string atr = line.c_str();
                lib[j][atr].insert(id);
            } else {
                getline(cin, line);
                istringstream iss(line);
                string key;
                while (iss >> key) {
                    lib[j][key].insert(id);
                }
            }
        }
    }

    cin >> m;
    for (i = 0; i < m; i++) {
        int num;
        scanf("%d: ", &num);
        string s;
        getline(cin, s);
        cout <<num << ": " << s << endl;
        if (lib[num].find(s) == lib[num].end()) {
            cout << "Not Found" << endl;
            continue;
        }
        for (set<int>::iterator it = lib[num][s].begin(); it != lib[num][s].end(); it++) {
            cout.fill('0');
            cout << setw(7) << *it << endl;
        }
    }

    return 0;
}