PAT甲题题解-1022. Digital Library (30)-map映射+vector

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <map>
#include <vector>
#define TITLE 1
#define AUTHER 2
#define KEYWORDS 3
#define PUBLISHER 4
#define YEAR 5
using namespace std;
const int maxn=10000+5;
int n;
int cnt=0;
map<string,int>maps[6];

struct Book{
    int id;
    string title;
    string author;
    string keywords[5];
    int keynum;
    string publisher;
    string year;
    bool operator<(const Book tmp)const{
        return id<tmp.id;
    }
}book[maxn];

vector<int> book_id[maxn*3];

void readKeywords(char*str,int i){
    //按照定义的分隔符d来分割字符串，对keyword进行读取
    const char *d = " ";//空格
    char *p;
    p = strtok(str,d);
    int idx=0;
    //这样做的主要原因是避免最后输出换行
    while(p)
    {
        book[i].keywords[idx]=p;
        idx++;
        p=strtok(NULL,d);  //读取下一个p
    }
    book[i].keynum=idx;
}
/**
建立k、string的映射
*/
void build_map(string s,int i,int k){
    if(maps[k][s]==0)
        maps[k][s]=++cnt;
    book_id[maps[k][s]].push_back(i);
}
int main()
{
    char str[100];
    scanf("%d",&n);
    for(int i=0;i<n;i++){
        scanf("%d",&book[i].id);
        getchar(); //这里是读取id后面的换行符
        gets(str);
        book[i].title=str;

        gets(str);
        book[i].author=str;

        gets(str);
        readKeywords(str,i);

        gets(str);
        book[i].publisher=str;

        gets(str);
        book[i].year=str;
    }
    sort(book,book+n);
    //建立映射，顺便存储书的id
    for(int i=0;i<n;i++){
        build_map(book[i].title,i,TITLE);
        build_map(book[i].author,i,AUTHER);
        for(int j=0;j<book[i].keynum;j++)
            build_map(book[i].keywords[j],i,KEYWORDS);
        build_map(book[i].publisher,i,PUBLISHER);
        build_map(book[i].year,i,YEAR);
    }
    int m;
    int kind;
    scanf("%d",&m);
    for(int i=0;i<m;i++){
        scanf("%d:",&kind);
        getchar();
        gets(str);
        printf("%d: %s\n",kind,str);
        string s=str;
        int map_id=maps[kind][s];
        if(map_id==0)
            printf("Not Found\n");
        else{
            for(int j=0;j<book_id[map_id].size();j++){
                int bid=book_id[map_id][j];
                printf("%07d\n",book[bid].id);
            }
        }
    }
    return 0;
}