[C++] PAT 1022 Digital Library (30分)

Sample Input:

3
1111111
The Testing Book
Yue Chen
test code debug sort keywords
ZUCS Print
2011
3333333
Another Testing Book
Yue Chen
test code sort keywords
ZUCS Print2
2012
2222222
The Testing Book
CYLL
keywords debug book
ZUCS Print2
2011
6
1: The Testing Book
2: Yue Chen
3: keywords
4: ZUCS Print
5: 2011
3: blablabla

Sample Output:

1: The Testing Book
1111111
2222222
2: Yue Chen
1111111
3333333
3: keywords
1111111
2222222
3333333
4: ZUCS Print
1111111
5: 2011
1111111
2222222
3: blablabla
Not Found

题解：

该题使用map<string,set<int>>将相同信息与id集对应

#include <iostream>
#include <string>
#include <map>
#include <set>

using namespace std;

int n;
map<string,set<int>> title,author,keywords,publisher,year;

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

}



int main(){

	cin >> n;

	int id;
	string t_title,t_author,t_key,t_publisher,t_year;
	for(int i=0;i<n;i++){
		cin >> id;
		getchar();

		getline(cin,t_title);
		title[t_title].insert(id);

		getline(cin,t_author);
		author[t_author].insert(id);

		while(cin >> t_key){
			keywords[t_key].insert(id);
		
			char c =getchar();

			if(c == '\n'){
				break;
			}
		
		}

		getline(cin,t_publisher);
		publisher[t_publisher].insert(id);

		getline(cin,t_year);
		year[t_year].insert(id);
	}
	
	int m;
	cin >> m;

	int num;
	string q;
	for(int i=0;i<m;i++){
		scanf_s("%d: ",&num);
		getline(cin,q);

		cout << num << ": " << q << endl;
		
		if(num == 1) query(title,q);
		else if(num == 2) query(author,q);
		else if(num == 3) query(keywords,q);
		else if(num == 4) query(publisher,q);
		else if(num == 5) query(year,q);
		
		

	}


	system("pause");
	return 0;
}