1022 Digital Library (30 分)

版本2

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

int main(){
	int n, m;
	cin >> n;
	string id, title, author, publisher, publishyear, key;
	getchar();
	unordered_map<string, set<string>> mp;
	for(int i = 0; i < n; i++){
		getline(cin, id);
		getline(cin, title);
		mp[title].insert(id);
		getline(cin, author);
		mp[author].insert(id);
		while(cin>>key){
			char c = getchar();
			mp[key].insert(id);
			if(c == '\n') break;
		}
		getline(cin, publisher);
		mp[publisher].insert(id);
		getline(cin, publishyear);
		mp[publishyear].insert(id);
	}
	cin >> m;
	int num;
	getchar();
	while(m--){
		getline(cin, key);
		num = key[0] - '0';
		key = key.substr(3);
		cout<<num<<": "<<key<<endl;
		if(mp[key].size() == 0){
			cout<<"Not Found"<<endl;
		}else{
			for(auto it: mp[key]){
				cout<<it<<endl;
			}
		}
	}
	return 0;
}

注意:
1.书名是7位数,需要补零。
2.题目要求按照从小到大输出书名,所以用set比较好。
3.关键字的读入技巧要学到。
4.对参数使用引用会稍微快点。

版本1

//map的应用 
#include
#include
#include
#include 
#include
using namespace std;
map<string,set<int> > mpT,mpA,mpK,mpP,mpY;
void query(map<string,set<int> >& mp, string &str){ //听说引用会快点 
	if(mp.find(str) == mp.end()) printf("Not Found\n");
	else{
		for(set<int>::iterator it=mp[str].begin(); it!=mp[str].end();it++){
			printf("%07d\n",*it);
		}
	}
}

int main(){
	int n,m,id,type;
	string title,author,key,pub,year;
	scanf("%d",&n);
	
	for(int i=0;i<n;i++){
		scanf("%d",&id);
		char c = getchar();
		getline(cin, title);
		mpT[title].insert(id);
		getline(cin,author);
		mpA[author].insert(id);
		while(cin>>key){
			mpK[key].insert(id);
			c = getchar();
			if(c == '\n') break;
		}
		getline(cin, pub);
		mpP[pub].insert(id);
		getline(cin,year);
		mpY[year].insert(id);
	}
	string temp;
	scanf("%d",&m);
	for(int i=0;i<m;i++){
		scanf("%d: ",&type);
		getline(cin, temp);
		cout<<type<<": "<<temp<<endl;
		if(type==1) query(mpT, temp);
		else if(type==2) query(mpA, temp);
		else if(type==3) query(mpK, temp);
		else if(type==4) query(mpP, temp);
		else query(mpY, temp);
	}	
	return 0;
} 

你可能感兴趣的:(STL,PAT甲级真题题解)