PAT乙级1065单身狗(利用set的特点）

1065 单身狗 (25 分)

“单身狗”是中文对于单身人士的一种爱称。本题请你从上万人的大型派对中找出落单的客人，以便给予特殊关爱。

输入格式：

输入第一行给出一个正整数 N（≤ 50 000），是已知夫妻/伴侣的对数；随后 N 行，每行给出一对夫妻/伴侣——为方便起见，每人对应一个 ID 号，为 5 位数字（从 00000 到 99999），ID 间以空格分隔；之后给出一个正整数 M（≤ 10 000），为参加派对的总人数；随后一行给出这 M 位客人的 ID，以空格分隔。题目保证无人重婚或脚踩两条船。

输出格式：

首先第一行输出落单客人的总人数；随后第二行按 ID 递增顺序列出落单的客人。ID 间用 1 个空格分隔，行的首尾不得有多余空格。

输入样例：

3
11111 22222
33333 44444
55555 66666
7
55555 44444 10000 88888 22222 11111 23333

输出样例：

5
10000 23333 44444 55555 88888

思路还是很简单，但要全部通过还是要运用好set这个数据结构

1.首先建立couplelist(配偶表），我是用了map，为了两个号码可以互查，所以key和value要互换再存一次

2.对于每一个号码，若在配偶表没有找到他的配偶，则他一定是单身狗

  若配偶表里他有对象，但对象没来，则也是单身狗，关键就是怎么样判断这种情况：我的思路建立一个set装入来的人，每一个人，查他的配偶插入set里，若插入成功（则说明set里没有配偶，仍为单身狗），若插入失败（set的特性，重复元素插入则失败），则说明来的人里有他的配偶，则他 不是单身狗。这样一次循环就可以判断所有人的单身情况，见代码

#include<iostream>
#include<string>
#include<set>
#include<map>
using namespace std;
int main(){
	int N,M;
	string msg,p1,p2,input;
	map<string, string>couplelist;
	set<string>st,st1;
	cin >> N;
	cin.ignore();
	for (int i = 0; i < N; i++){ //建立map，每个配偶都可以互相查对方号码
		getline(cin, msg);
		p1 = msg.substr(0, 5);
		p2 = msg.substr(6, 5);
		couplelist[p1] = p2;
		couplelist[p2] = p1;
	}
	cin >> M;
	vector<string>costoms(M);
	cin.ignore();
	getline(cin, input);
	int index = 0,flag=0;
	for (int i = 0; i < M; i++){ //从字符串中提取每个号码段存入costoms[i]数组里
		for (int j = index; j < input.length(); j++)
		{
			if (input[j] == ' '){
				index = j + 1;
				break;
			}
			costoms[i].push_back(input[j]);
		}
		st1.insert(costoms[i]);  //建立比对库
	}
	string search;
	for (int i = 0; i < M; i++){
		search = couplelist[costoms[i]];
		if (search == ""){ //库里找不到的一定是单身狗
			st.insert(costoms[i]);
			continue;
		}
		auto t = st1.insert(search); //库里有配对，判断来的人里有没有配偶
		if (t.second == true){ //插入成功则st1中没有该元素，则配偶没有来，则判断单身
			st.insert(costoms[i]);
		}
	}
	cout << st.size() << endl;
	int k = 0;
	for (auto it : st){
		if (k != 0)cout << " ";
		k++;
		cout << it;
	}
	return 0;
}