PAT (Advanced Level) 1022. Digital Library (30)

简单模拟题。

写的时候注意一些小优化，小心TLE。

#include<iostream>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<cstdio>
#include<map>
#include<queue>
#include<vector>
using namespace std;

struct X
{
    string id;
    char title[]; int Title;
    char author[]; int Author;
    vector<int>key;
    char publisher[]; int Publisher;
    int year;
}s[+];
int n,m;

map<string,int>Tit,Aut,Key,Pub;
vector<string>ans;
int tot_Tit=,tot_Aut=,tot_Key=,tot_Pub=;

bool cmp(const X&a,const X&b)
{
    return a.id<b.id;
}

int main()
{
    scanf("%d",&n);
    for(int i=;i<=n;i++)
    {
        cin>>s[i].id; getchar();
        gets(s[i].title); if(Tit[s[i].title]==) Tit[s[i].title]=++tot_Tit; s[i].Title=Tit[s[i].title];
        gets(s[i].author); if(Aut[s[i].author]==) Aut[s[i].author]=++tot_Aut; s[i].Author=Aut[s[i].author];
        char tmp[]; gets(tmp);
        int len=strlen(tmp);
        string q;
        for(int j=;j<=len;j++)
        {
            if(tmp[j]==' '||tmp[j]=='\0')
            {
                if(Key[q]==) Key[q]=++tot_Key;
                s[i].key.push_back(Key[q]);
                q.clear();
            }
            else q=q+tmp[j];
        }
        gets(s[i].publisher); if(Pub[s[i].publisher]==) Pub[s[i].publisher]=++tot_Pub; s[i].Publisher=Pub[s[i].publisher];
        scanf("%d",&s[i].year);
    }

    sort(s+,s++n,cmp);

    scanf("%d",&m);
    for(int i=;i<=m;i++)
    {
        char op[]; scanf("%s",op); getchar();
        char h[]; gets(h);
        printf("%s %s\n",op,h);

        if(op[]=='')
        {
            int num=Tit[h];
            for(int i=;i<=n;i++)
                if(s[i].Title==num) ans.push_back(s[i].id);
        }
        else if(op[]=='')
        {
            int num=Aut[h];
            for(int i=;i<=n;i++)
                if(s[i].Author==num) ans.push_back(s[i].id);
        }
        else if(op[]=='')
        {
            int num=Key[h];
            for(int i=;i<=n;i++)
            {
                for(int j=;j<s[i].key.size();j++)
                {
                    if(s[i].key[j]==num)
                    {
                        ans.push_back(s[i].id);
                        break;
                    }
                }
            }
        }
        else if(op[]=='')
        {
            int num=Pub[h];
            for(int i=;i<=n;i++)
                if(s[i].Publisher==num) ans.push_back(s[i].id);
        }
        else if(op[]=='')
        {
            int num=;
            for(int i=;h[i];i++) num=num*+h[i]-'';
            for(int i=;i<=n;i++)
                if(s[i].year==num) ans.push_back(s[i].id);
        }
        if(ans.size()==) printf("Not Found\n");
        else
        {
            for(int i=;i<ans.size();i++) cout<<ans[i]<<endl;
            ans.clear();
        }
    }
    return ;
}