1. 程式人生 > >PAT 1065 單身狗

PAT 1065 單身狗

set () scanf 數字 愛稱 復雜度 sort == 題目

https://pintia.cn/problem-sets/994805260223102976/problems/994805266942377984

“單身狗”是中文對於單身人士的一種愛稱。本題請你從上萬人的大型派對中找出落單的客人,以便給予特殊關愛。

輸入格式:

輸入第一行給出一個正整數 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

時間復雜度:$O(N * logN)$

代碼:

#include <bits/stdc++.h>
using namespace std;

const int maxn = 1e5 + 10;
int a[10010], vis[maxn], out[10010];

int main() {
    int N;
    scanf("%d", &N);
    map<int, int> mp;
    mp.clear();
    for(int i = 1; i <= N; i ++) {
        int x, y;
        scanf("%d%d", &x, &y);
        mp[x] = y;
        mp[y] = x;
    }

    int M;
    scanf("%d", &M);
    memset(vis, 0, sizeof(vis));
    for(int i = 1; i <= M; i ++) {
        scanf("%d", &a[i]);
        vis[a[i]] = 1;
    }

    int num = 0;
    bool flag = false;
    for(int i = 1; i <= M; i ++) {
        if(vis[mp[a[i]]] == 0) {
            num ++;
            out[num] = a[i];
        }
    }

    sort(out + 1, out + num + 1);
    printf("%d\n", num);
    for(int i = 1; i <= num; i ++) {
        printf("%05d", out[i]);
        printf("%s", i != num ? " " : "\n");
    }
    return 0;
}

  

PAT 1065 單身狗