1022 Digital Library (30 分)

1022 Digital Library (30 分)

题意

给出n本书的信息，包括id（由7位数字组成），书名（由字符串组成），作者（由字符串组成），关键字（由字符串组成），出版社（由字符串组成），出版年限（由4位数字组成）
给出m个询问，询问有五种情况，分别有问书名，作者，关键字，出版社，出版年限，要求输出符合询问的书籍的编号。（如问作者，输出所有以该作者为作者的书的编号）
n<=1e4
m<=1e3

思路

用map对所有书名，作者，关键字，出版社，出版年限关联一组set，set中记录数的id。
按要求输入输出即可。
注意map关联set的这种写法，以及一组用空格隔开的字符串的输入方法。

代码

#include<stdio.h>
#include<map>
#include<string>
#include<iostream>
#include<set>

using namespace std;

void query(map<string,set<int> >& mp,string& str)
{
	if(mp.find(str)!=mp.end()){
		for(set<int>::iterator it=mp[str].begin();it!=mp[str].end();it++){
			printf("%07d\n",*it);
		}
	}else{
		printf("Not Found\n");
	}
}

int main()
{
	map<string,set<int> >title,author,keyWord,publisher,year;
	int n,id,m,type;
	string Title,Author,KeyWord,Publisher,Year;
	char ch;
	string str;
	scanf("%d",&n);
	for(int i=0;i<n;i++){
		scanf("%d",&id);
		getchar();
		getline(cin,Title);
		title[Title].insert(id);
		getline(cin,Author);
		author[Author].insert(id);
		while(cin>>KeyWord){
			keyWord[KeyWord].insert(id);
			ch=getchar();
			if(ch=='\n'){
				break;
			}
		}
		getline(cin,Publisher);
		publisher[Publisher].insert(id);
		getline(cin,Year);
		year[Year].insert(id);
	}
	scanf("%d",&m);
	for(int i=0;i<m;i++){
		scanf("%d: ",&type);
		getline(cin,str);
		cout<<type<<": "<<str<<endl;
		if(type==1){
			query(title,str);
		}else if(type==2){
			query(author,str);
		}else if(type==3){
			query(keyWord,str);
		}else if(type==4){
			query(publisher,str);
		}else{
			query(year,str);
		}
	}
	return 0;
}