PAT(B) 1065 单身狗（Java：17分，C：25分）

题目链接：1065 单身狗 (25 point(s))

题目描述

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

输入格式

输入第一行给出一个正整数 N（≤ 50 000），是已知夫妻/伴侣的对数；随后 N 行，每行给出一对夫妻/伴侣——为方便起见，每人对应一个 ID 号，为 5 位数字（从 00000 到 99999），ID 间以空格分隔；之后给出一个正整数 M（≤ 10 000），为参加派对的总人数；随后一行给出这 M 位客人的 ID，以空格分隔。题目保证无人重婚或脚踩两条船。

输出格式

首先第一行输出落单客人的总人数；随后第二行按 ID 递增顺序列出落单的客人。ID 间用 1 个空格分隔，行的首尾不得有多余空格。

测试样例

Case 0:

3
11111 22222
33333 44444
55555 66666
7
55555 44444 10000 88888 22222 11111 23333

5
10000 23333 44444 55555 88888

Case 1:

3
11111 22222
33333 44444
55555 66666
4
55555 44444 66666 33333

0

分析?

Java 和 C 代码思路一样，但是 Java 有两组数据超时。

• 一个数组保存对象的 ID
• 一个数组保存到场的情况
• 一个数组保存单身狗
• 没有对象的，对象没到场的都是单身狗

Java代码（17分）

/*
Case 3: TLE
Case 4: TLE
*/
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;

public class Main {
	private static final int MAX_ID = 99999;

	private static BufferedReader input;
	private static int coupleNumber;// 夫妻对数
	private static int[] coupleID;// 丈夫/妻子的 ID
	private static int totalPeople;// 派对总人数
	private static int[] guestID;// 客人 ID
	private static boolean[] flagPresent;// 到场标志

	private static void initBufferedReader() {
		InputStreamReader isr = new InputStreamReader(System.in);
		input = new BufferedReader(isr);
	}

	private static void inputData() {
		initCoupleNumber();
		initCoupleID();
		initTotalPeople();
		initGuestID();
		initFlagPresent();
	}

	private static void initCoupleNumber() {
		String str = "0";
		try {
			str = input.readLine();
		} catch (IOException e) {
		}
		coupleNumber = Integer.parseInt(str);// 夫妻对数
	}

	private static void initCoupleID() {
		coupleID = new int[MAX_ID + 1];
		for (int i = 0; i <= MAX_ID; ++i) {
			coupleID[i] = -1;
		}

		for (int i = 0; i < coupleNumber; ++i) {
			String str = "";
			try {
				str = input.readLine();
			} catch (IOException e) {
			}
			String[] arr = str.split(" ");
			int aID = Integer.parseInt(arr[0]);
			int bID = Integer.parseInt(arr[1]);
			coupleID[aID] = bID;
			coupleID[bID] = aID;
		}
	}

	private static void initTotalPeople() {
		String str = "0";
		try {
			str = input.readLine();
		} catch (IOException e) {
		}
		totalPeople = Integer.parseInt(str);
	}

	private static void initGuestID() {
		guestID = new int[totalPeople];

		String str = "";
		try {
			str = input.readLine();
		} catch (IOException e) {
		}
		String[] arr = str.split(" ");

		for (int i = 0; i < totalPeople; ++i) {
			guestID[i] = Integer.parseInt(arr[i]);
		}

		Arrays.sort(guestID);
	}

	private static void initFlagPresent() {
		flagPresent = new boolean[MAX_ID + 1];
		for (int i = 0; i < totalPeople; ++i) {
			flagPresent[guestID[i]] = true;
		}
	}

	private static int[] getSingleDogs() {
		int[] singleDogs = new int[totalPeople];
		int singleNumber = 0;
		for (int i = 0; i < totalPeople; ++i) {
			int myID = guestID[i];
			int partnerID = coupleID[myID];
			if (partnerID < 0 || !flagPresent[partnerID]) {
				singleDogs[singleNumber++] = myID;
			}
		}
		return Arrays.copyOfRange(singleDogs, 0, singleNumber);
	}

	private static void outputSingleDogs(int[] singleDogs) {
		int length = singleDogs.length;
		System.out.println(length);
		System.out.printf("%05d", singleDogs[0]);
		for (int i = 1; i < length; ++i) {
			System.out.printf(" %05d", singleDogs[i]);
		}
	}

	public static void main(String[] args) {
		initBufferedReader();
		inputData();

		int[] singleDogs = getSingleDogs();
		outputSingleDogs(singleDogs);
	}
}

C代码

/*****************************************************************************
Submit Time			Status		Score	Problem	Compiler	Run Time	User
8/27/2019, 19:49:25	Accepted	25		1065	C (gcc)		13 ms		wowpH
*****************************************************************************/
#include<stdio.h>
#include<memory.h>									// memset头文件
#include<stdlib.h>									// qsort头文件

#define MAX_GUEST 100000							// 最大客人人数

int coupleNumber;									// 夫妻对数
int totalPeople;									// 派对总人数
int partnerID[MAX_GUEST];							// 对象ID
int guest[MAX_GUEST];								// 客人
int flagPresent[MAX_GUEST];							// 是否到场

int singleDogs[MAX_GUEST];							// 单身狗
int singleDogsNumber;								// 单身狗人数

void initData();									// 初始化数据
void inputData();									// 输入数据
int cmp(const int* a, const int* b);				// 比较函数
void processData();									// 处理部分数据
int checkSingleDog(int id);							// 检查id是否是单身狗
void setSingleDogs();								// 计算单身狗
void outputSingleDogs();							// 输出单身狗

int main() {
	initData();
	inputData();
	processData();
	setSingleDogs();
	outputSingleDogs();
	return 0;
}

void initData() {
	memset(partnerID, -1, sizeof(partnerID));
	memset(flagPresent, -1, sizeof(flagPresent));
}

void inputData() {
	scanf("%d", &coupleNumber);

	int id1, id2;
	for (int i = 0; i < coupleNumber; ++i) {
		scanf("%d %d", &id1, &id2);
		partnerID[id1] = id2;
		partnerID[id2] = id1;
	}

	scanf("%d", &totalPeople);

	for (int i = 0; i < totalPeople; ++i) {
		scanf("%d", &guest[i]);
	}
}

int cmp(const int* a, const int* b) {					// 升序
	return (*a) - (*b);
}

void processData() {
	qsort(guest, totalPeople, sizeof(int), cmp);		// 客人排序

	for (int i = 0; i < totalPeople; ++i) {
		flagPresent[guest[i]] = 1;						// 设置客人到场标志
	}
}

void setSingleDogs() {
	singleDogsNumber = 0;
	for (int i = 0; i < totalPeople; ++i) {
		int self = guest[i];
		if (checkSingleDog(self) > 0) {
			singleDogs[singleDogsNumber++] = guest[i];	// 是单身狗
		}
	}
}

int checkSingleDog(int id) {
	int partner = partnerID[id];
	if (partner < 0) {									// 没有对象
		return 1;
	} else if (flagPresent[partner] < 0) {				// 有对象，但没到场
		return 1;
	}
	return -1;
}

void outputSingleDogs() {
	printf("%d\n", singleDogsNumber);
	for (int i = 0; i < singleDogsNumber; ++i) {
		if (i > 0) {
			printf(" ");
		}
		printf("%05d", singleDogs[i]);
	}
}

提交结果（C）