PTA L1-020 帅到没朋友(Java)

2023-08-22 13:59:28

L1-020 帅到没朋友 (20 分)

当芸芸众生忙着在朋友圈中发照片的时候，总有一些人因为太帅而没有朋友。本题就要求你找出那些帅到没有朋友的人。

输入格式：

输入第一行给出一个正整数N（≤100），是已知朋友圈的个数；随后N行，每行首先给出一个正整数K（≤1000），为朋友圈中的人数，然后列出一个朋友圈内的所有人——为方便起见，每人对应一个ID号，为5位数字（从00000到99999），ID间以空格分隔；之后给出一个正整数M（≤10000），为待查询的人数；随后一行中列出M个待查询的ID，以空格分隔。

注意：没有朋友的人可以是根本没安装“朋友圈”，也可以是只有自己一个人在朋友圈的人。虽然有个别自恋狂会自己把自己反复加进朋友圈，但题目保证所有K超过1的朋友圈里都至少有2个不同的人。

输出格式：

按输入的顺序输出那些帅到没朋友的人。ID间用1个空格分隔，行的首尾不得有多余空格。如果没有人太帅，则输出No one is handsome。

注意：同一个人可以被查询多次，但只输出一次。

输入样例1：

3
3 11111 22222 55555
2 33333 44444
4 55555 66666 99999 77777
8
55555 44444 10000 88888 22222 11111 23333 88888
结尾无空行

输出样例1：

10000 88888 23333
结尾无空行

输入样例2：

3
3 11111 22222 55555
2 33333 44444
4 55555 66666 99999 77777
4
55555 44444 22222 11111

输出样例2：

No one is handsome

采用集合的做法，最后一个测试点总是运行超时，怀疑是时间复杂度过高，但并不想牺牲内存。。XD

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.Iterator;
import java.util.LinkedHashSet;

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());
        ArrayList<String> arrayList = new ArrayList<>();
        for (int i = 0; i < N; i++) {
            String[] strings = br.readLine().split(" ");
            int num = Integer.parseInt(strings[0]);
            if(num == 1|| num == 0)
                continue;
            for (int j = 1; j < strings.length; j++) {
                arrayList.add(strings[j]);
            }
        }
        int cnt = Integer.parseInt(br.readLine());
        String[] man = br.readLine().split(" ");
        LinkedHashSet<String> set = new LinkedHashSet<>();
        for (int i = 0; i < cnt; i++) {
            if(!arrayList.contains(man[i])){
                set.add(man[i]);
            }
        }
        Iterator it = set.iterator();
        int c = 0;
        if(set.size()!=0) {
            while (it.hasNext()) {
                c++;
                if (c != set.size()) {
                    System.out.print(it.next() + " ");
                } else
                    System.out.print(it.next());
            }
        }
        else
            System.out.println("No one is handsome");
    }
}

码农公寓