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

思路:
数组下标法!!!
couple数组的下标存储id,每次输完一对夫妇的id后,用temp给couple数组相应的id赋相同的值,表明这是一对夫妻(每对夫妻temp值不同),如果任意两个数组的值相同表明他们是夫妻关系!
输入完所有数据后:
对客人数组按其相对应的couple数组的temp 的值降序排序(升序也行无所谓)。
因为结果需要升序排序,所以采用set

如果某客人的couple数组为0 或 前后不相等均是无伴侣的,反之两个是伴侣,i需要 + 2

题解:

#include<iostream>
#include<algorithm>
#include<set>

using namespace std;

const int N = 1e6, M = 1e4;
int couple[N], guest[M];

bool cmp(int a, int b) {
    
     return couple[a] > couple[b]; }

int main() {
    
    
	ios::sync_with_stdio(false);
	int n, m;
	int x, y, temp = 1;
	scanf("%d", &n);
	for (int i = 0; i < n; i++) {
    
    
		scanf("%d %d", &x, &y);
		couple[x] = couple[y] = temp++;
	}
	scanf("%d", &m);
	for (int i = 0; i < m; i++) scanf("%d", &guest[i]);
	//核心步骤
	sort(guest, guest + m, cmp);
	set<int>s;
	int i = 0;
	while (i < m) {
    
    
		if (couple[guest[i]] == 0) s.insert(guest[i++]);
		else if (couple[guest[i]] != couple[guest[i + 1]]) s.insert(guest[i++]);
		else i += 2;
	}
	//输出结果
	printf("%d\n", s.size());
	bool flag = false;
	for (int x : s) {
    
    
		printf("%s%.5d", flag ? " " : "", x);//巧妙!!!!
		flag = true;
	}

	return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_44715943/article/details/115176914