【PAT乙级】1065 单身狗 (25分)

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

输入格式：

输入第一行给出一个正整数 N（$\le$ 50 000），是已知夫妻/伴侣的对数；随后 N 行，每行给出一对夫妻/伴侣——为方便起见，每人对应一个 ID 号，为 5 位数字（从 00000 到 99999），ID 间以空格分隔；之后给出一个正整数 M（$\le$ 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

解题思路：

• 两个 map 分别是用于保存配偶映射的 couple 和出席人员 occur，检查配偶是否出现过，配偶出现过，则夫妻双方均置false。
• 向量 v 按顺序保存出席人员 id，集合 s 根据 occur[ v[id] ] 保存单身狗 id。

答案代码：

#include <iostream>
#include <string>
#include <vector>
#include <map>
#include <set>
using namespace std;

map<string, string> couple;
map<string, bool> occur;

int main() {
	/*	1. 配偶映射 couple		*/ 
    int n = 0;
    cin >> n;

	string male, female;
    for (int i = 0; i < n; i++) {
    	cin >> male >> female;
        couple[male] = female;  couple[female] = male;
    }
    
    /*	2. 出席人员映射 occur	*/ 
    int m = 0;
    cin >> m;
    
    string id;
    vector<string> v;
    for (int i = 0; i < m; i++) {
        cin >> id;
        v.push_back(id);							// 出席人员向量 
        
        occur[id] = true;						
        if (occur[couple[id]]) {                	// 配偶出现过，则夫妻双方均置false 
            occur[id] = false;  occur[couple[id]] = false;
        }
    }
    
    /* 	3. 单身狗集合 set	*/ 
    set<string> s;
    for (int i = 0; i < m; i++) 
        if (occur[v[i]]) s.insert(v[i]);              // 按照出席顺序存储 
    cout << s.size() << endl;
    
    for (auto it = s.begin(); it != s.end(); it++) {
        if (it != s.begin()) cout << " ";
        cout << *it;
    }
    return 0;
}