PAT甲1022 Digital Library (30)（30 分）

#include 
#include 
#include 
#include 
#include 
#include 
#include 
using namespace std;

int N;
struct book
{
    string id;
    string sindex[10];
}b[10010];

set<string> keyword[10100];
map<string,int> mp;
int num=0;

bool cmp(book a, book b)
{
    return a.idvoid split_s(string str,string id)
{
    string temp="";
    int pre=0;
    for(int i=0;iif(str[i]==' '||i==str.length()-1)
        {
            if(str[i]==' ')
            {
              temp=str.substr(pre,i-pre);
             }
             else
             {
                temp=str.substr(pre,i-pre+1);
             }
            pre=i+1;
            if(mp.find(temp)==mp.end())
            {
                mp[temp]=num++;
            }
            int ki=mp[temp];
            keyword[ki].insert(id);
        }
    }
}

int main()
{
    cin>>N;
    getchar();
    for(int i=0;istring id;
        getline(cin,id);
        b[i].id=id;
        for(int j=1;j<=5;j++)
        {
            getline(cin,b[i].sindex[j]);
        }
        split_s(b[i].sindex[3],id);
    }
    sort(b,b+N,cmp);
    int K;
    scanf("%d",&K);
    for(int i=0;iint qi;
        string qkey="";
        scanf("%d: ",&qi);
        getline(cin,qkey);
        cout<": "<bool flag=false;
        if(qi!=3)
        {
            for(int j=0;jif(b[j].sindex[qi]==qkey)
                {
                    cout<true;
                }
            }
        }
        else
        {
            if(mp.find(qkey)!=mp.end())
            {
                int ki=mp[qkey];
                for(set<string>::iterator it= keyword[ki].begin();it!=keyword[ki].end();it++)
                {
                    cout<<*it<true;
                }
            }
        }
        if(!flag)
        {
            cout<<"Not Found"<return 0;
}