个人博客:小景哥哥
1065.单身狗(25)–PAT乙级真题java实现
“单身狗”是中文对于单身人士的一种爱称。本题请你从上万人的大型派对中找出落单的客人,以便给予特殊关爱。
输入格式:
输入第一行给出一个正整数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
分析:
设立数组couples[i] = j表示i的对象是j。一开始先设置为都是-1。设立数组appear表示某人的对象是否来到了派对上。接收数据的时候,对于每一对wife和husband,将couple的husband设置为wife,wife设置为husband,表示他俩是一对。对于每一个需要判断的人,将其存储在guest数组里面,如果它不是单身的(也就是如果它的couple[guest[i]] != -1)那么就将它对象的appear设置为1,表示他对象的对象(也就是他自己)来到了派对。这样所有appear不为1的人,对象是没有来到派对的。
//后两个测试点超时,宝宝已经尽力try过了
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;
public class Main {
public static void main(String[] args) throws IOException{
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
int N = Integer.parseInt(br.readLine());
int couples[] = new int[100000];
int appear[] = new int[100000];
for(int i = 0; i < 100000; i++) {
couples[i] = -1;
appear[i] = -1;
}
for(int i = 0; i < N; i++) {
String[] s = br.readLine().split(" ");
int wife = Integer.parseInt(s[0]);
int husband = Integer.parseInt(s[1]);
couples[wife] = husband;
couples[husband] = wife;
}
int M = Integer.parseInt(br.readLine());
String[] guest = br.readLine().split(" ");
int minus = 0;
for(int i = 0; i < M; i++) {
int g = Integer.parseInt(guest[i]);
if(couples[g] != -1)
appear[couples[g]] = 1;
if(couples[g] != -1 && appear[couples[g]] == 1 && appear[g] == 1) {
minus += 2;
}
}
Arrays.sort(guest);
boolean flag = false;
System.out.println(guest.length - minus);
for(int i = 0; i < guest.length;i++) {
int g = Integer.parseInt(guest[i]);
if(appear[g] != 1) {
if(flag) {
System.out.print(" " + guest[i]);
}else {
System.out.print(guest[i]);
flag = true;
}
}
}
}
}