PAT：1022. Digital Library (30) （部分正确，错最后两个）

#include<stdio.h>
#include<string.h>
#include<map>
#include<string>
#include<set>
#include<iostream>
using namespace std;
int n;        //n本书
//标题、作者、关键字、出版社、出版年份与ID的映射
map<string,set<int>> mpTITLE,mpAUTHOR,mpKEY,mpPUBLISHER,mpYEAR;

void query(map<string,set<int>> &m,string Q)    //【skill】“&m”的引用符号是为了加快处理速度
{
    if(m.find(Q)==m.end())
        printf("Not Found\n");
    else
        for(set<int>::iterator it=m[Q].begin() ; it!=m[Q].end() ; ++it)    //【思维】m已经选好了是作者还是出版社之类的，it只要扫描这个类的set，按个输出就可以了
            printf("%d\n",*it);                //【skill】it所指向的内容
}

int main()
{
    scanf("%d",&n);
    int ID=-1;
    string title,author,key,publisher,year;
    for(int i=0 ; i<n ; ++i)
    {
        scanf("%d",&ID);
        char c=getchar();            //【caution】吸收换行符
        getline(cin,title);            //输入标题
        mpTITLE[title].insert(ID);
        getline(cin,author);
        mpAUTHOR[author].insert(ID);
        while(cin>>key)                //【skill】持续输入关键字的方法，cin也是按空格划分的
        {
            mpKEY[key].insert(ID);
            char d=getchar();        //【skill】取后面的字符，是空格，则还有key，是“\n则输入结束，break”
            if(d=='\n')
                break;
        }
        getline(cin,publisher);
        mpPUBLISHER[publisher].insert(ID);
        getline(cin,year);
        mpYEAR[year].insert(ID);
    }
    int q,num;
    scanf("%d",&q);
    for(int i=0 ; i<q ; ++i)
    {
        scanf("%d: ",&num);        //看看这样可否
        string Q;
        getline(cin,Q);
        printf("%d: %s\n",num,Q.c_str());    //【skill】将Q转为可用printf输出的%s
        if(1==num)
            query(mpTITLE,Q);
        else if(2==num)
            query(mpAUTHOR,Q);
        else if(3==num)
            query(mpKEY,Q);
        else if(4==num)
            query(mpPUBLISHER,Q);
        else if(5==num)
            query(mpYEAR,Q);
 

    }
    return 0;
}