PAT 1065 单身狗（25）（STL-map+思路+测试点分析）

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

PS：看到这题目，当场去世！！！！

我的思路：首先录入伴侣的信息，然后当输入参加派对的那些人时，把他的key值置1，然后遍历一遍所有人，当满足单身狗条件时（本身key为1、伴侣的key为0），记录这组数据。随后输出就行。（显然我这里最后存储的时候用map就有点大材小用了，其实用set就行了，懒得加头文件和定义了）

注意：1、考虑当count为0的情况，否则测试点2会出现段错误，运行超时；

2、注意时间复杂度要低，否则测试点3、4运行超时。

#include<iostream>
#include<map>
#include<string>
using namespace std;
int main(){
	map<string, pair<string,int>> s, res;     //s用于判断是否单身
	map<string, pair<string,int>>::iterator it;
	int n, m, count = 0;
	string r, l;
	cin >> n;
	while (n--) {
		cin >> l >> r;
		s[l].first = r;		//存入伴侣信息
		s[r].first = l;
	}
	cin >> m;
	for (int i = 0; i < m; i++) {
		cin >> l;
		s[l].second++;
	}
	for (it = s.begin(); it != s.end(); it++) {
		if (it->second.second && !s[it->second.first].second) {  //逻辑判断：自己出席了会议，伴侣没有
			count++;               //记录单身狗数目
			res[it->first].second++;    //将符合条件的人存到res中
		}
	}
	cout << count << endl;
	if (count) {        //注意count为0时，就不用输出ID了，否则测试点2出现段错误、运行超时
		it = res.begin();
		cout << it++->first;
		for (; it != res.end(); it++) {
			cout << " " << it->first;
		}
	}
	return 0;
}