PTA:7-100 单身狗 (25分)---加解析

7-100 单身狗 (25分)

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

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

思路
该题我的思路主要是利用数组下标的技巧，来进行标记。
具体解析见代码：

#include <bits/stdc++.h>
using namespace std;
int p[100010],ans[100010];
 
int main(){
	int n, a, b;
	memset(p, -1, sizeof(p));  //-1表示单身 
	cin >> n;
	while(n--){
		cin >> a >> b;
		p[a]=b;  //存下情侣其对方的编号 
		p[b]=a;
	}
	int m, c;
	cin >> m;
	while(m--){
		cin >> c;
		if(p[c]==-1) {  //肯定是单身 
			ans[c] = 1;
		}else{
			if(p[p[c]]==-2){  //p[c]值为伴侣编号 
				ans[p[c]]=0; 
			}else{  //可能是单身 
				ans[c] = 1;   
				p[c] = -2;  //将其标记一下，因为不确定后面还有没有其伴侣 
			} 	
		}
	}
	int cnt=0,flag=0;
	for(int i=0; i<100010; i++){
		if(ans[i]) cnt++;  //统计个数 
	}
	cout << cnt << endl;
	for(int i=0; i<100010; i++){
		if(ans[i]){
			if(flag) cout << " ";
			cout << setw(5) << setfill('0') << i;  //方式出现编号为1的情况，进行补零即：00001 
			flag=1;
		} 
	}
	return 0;
}

欢迎大家批评改正！！！