【PAT乙级】1065 单身狗 (25分 C++实现 暴力的话一个简单题)

题目

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

输入格式：
输入第一行给出一个正整数 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

思路

给的时间限制不大，而且题目数据量也很少，我就直接暴力做了，按照题目的意思，把每个来参加的人遍历一遍看看有没有ta的配偶就行了，注意点一个是配偶是无向的，所以两个方向都要，还有一个是输出的时候00001之类的不要漏了0（测试点4）。

我这个是纯暴力写法，比较简单易懂，其中能优化的地方非常多，至少可以优化成O（mlog（m））的，应该还有更好的写法，后面我想一想再更

代码

#include<bits/stdc++.h>
using namespace std;
int main() {//暴力的
	int n, x, m, cnt = 0, flag;
	vector<int >a(100001);
	cin >> n;
	for(int i = 0; i < n; i++) {
		cin >> x;
		cin >> a[x]; 
		a[a[x]] = x;
	}
	cin >> m;
	int b[m + 5], c[m + 5];
	for(int i = 0; i < m; i++) cin >> b[i];
	for(int i = 0; i < m; i++) {
		flag = 0;
		for(int j = 0; j < m; j++) {
			if(b[j] == a[b[i]]) {
				flag = 1;
				break;
			}
		}
		if(!flag) c[cnt++] = b[i];
	}
	cout << cnt << endl;
	sort(c, c + cnt);
	for(int i = 0; i < cnt; i++){
		if(i) cout << " ";
		printf("%05d", c[i]);
	}
	return 0;
}

优化1

优化记录了一下配偶是否到来，减少了遍历次数，不用每次都遍历

更好的方法的话是用set来做，就是查询到配偶存在之后，直接使用set容器的删除就行了，不用多一个for循环查询

#include<bits/stdc++.h>
using namespace std;
int main() {
	int n, x, m, cnt = 0, flag;
	vector<int >a(100001);
	cin >> n;
	for(int i = 0; i < n; i++) {
		scanf("%d", &x);
		scanf("%d", &a[x]);
		a[a[x]] = x;
	}
	cin >> m;
	int b[m + 5], c[m + 5];
	for(int i = 0; i < m; i++) cin >> b[i];
	for(int i = 0; i < m; i++) {
		flag = 0;
		for(int j = 0; j < m; j++) {
			if(b[j] == a[b[i]]) {
				flag = 1;
				break;
			}
		}
		if(!flag) c[cnt++] = b[i];
	}
	cout << cnt << endl;
	sort(c, c + cnt);
	for(int i = 0; i < cnt; i++) {
		if(i) cout << " ";
		printf("%05d", c[i]);
	}
	return 0;
}

优化2

set做的优化2

#include<bits/stdc++.h>
using namespace std;
int main() {
	int n, x, m, flag;
	vector<int >a(100001);
	vector<int >c(100001, 0);
	cin >> n;
	for(int i = 0; i < n; i++) {
		scanf("%d", &x);
		scanf("%d", &a[x]);
		a[a[x]] = x;
	}
	cin >> m;
	set<int >b;
	for(int i = 0; i < m; i++) {
		cin >> x;
		c[x] = 1;
		if(c[a[x]]) b.erase(a[x]);
		else b.insert(x);
	}
	cout << b.size() << endl;
	for(auto it = b.begin(); it != b.end(); it++) {
		if(it != b.begin()) cout << " ";
		printf("%05d", *it);
	}
	return 0;
}