1022. Digital Library (30)

这段题目采用了倒排索引的思路，为每一个需要查找的关键字建立一个map，其中的值存放的是书的ID。因为关键字分为5种，所以建立了map数组。同时一个关键字所对应的书的ID可能有多个，所以使用vector来存储ID。

#include<iostream>
#include<string>
#include<vector>
#include<cstdio>
#include<map>
#include<algorithm>

using namespace std;

map<string,vector<string> > bookmap[6];
int N,M;

int main()
{

    string id;
    string title;
    string author;
    string keyword;
    string publisher;
    string year;
    cin >> N;
    getchar();//输入N的时候，要将后面的回车过滤掉。
    for(int i = 0 ; i < N;i++)
    {
        char c;
        //getchar();
        getline(cin,id);
        getline(cin,title);
        bookmap[1][title].push_back(id);
        getline(cin,author);
        bookmap[2][author].push_back(id);
        do{
            cin >> keyword;
            c = getchar();
            bookmap[3][keyword].push_back(id);
        }while(c != '\n');
        getline(cin,publisher);
        bookmap[4][publisher].push_back(id);
        getline(cin,year);
        bookmap[5][year].push_back(id);
    }

    for(int i = 1; i < 6; ++i)
    {
        map<string,vector<string> >::iterator it;
        for(it = bookmap[i].begin(); it != bookmap[i].end(); it++)
        {
            sort(it->second.begin(), it->second.end());
        }
    }

    cin >> M;
    int tag;
    string str;
    for(int i = 0; i < M;i++)
    {
        scanf("%d: ",&tag);
        getline(cin,str);
        printf("%d: ",tag);
        cout << str << endl;
        map<string,vector<string> >::iterator it;
        it = bookmap[tag].find(str);
        if(it!=bookmap[tag].end())
        {
            vector<string> result = bookmap[tag][str];
            for(int i = 0; i < result.size(); ++i)
            {
                cout<<result[i]<<endl;
            }
        }else{
            printf("Not Found\n");
        }
    }

    return 0;
}

这道题如果ID采用int来存储，导致有几个测试案例过不了。