PAT BASIC LEVEL 1065. 单身狗(25)

1065. 单身狗(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

Answer:

#include<iostream>
using namespace std;
void print_id(int id, bool blank) {
    if(blank) cout << ' ';
    if(id < 10)
        cout << "0000";
    else if(id < 100)
        cout << "000";
    else if(id < 1000)
        cout << "00";
    else if(id < 10000)
        cout << '0';
    cout << id;
}

int main() {
    int n, m, t, single, party[100000] = {};
    cin >> n;
    int cps[n][2];
    for(int i=0; i<n; i++)
        cin >> cps[i][0] >> cps[i][1];
    cin >> m;
    single = m;
    while(m--) {
        cin >> t;
        party[t] = 1;
    }
    for(int i=0; i<n; i++)
        if(party[cps[i][0]] && party[cps[i][1]]) {
            party[cps[i][0]] = 0;
            party[cps[i][1]] = 0;
            single -= 2;
        }
    cout << single << '\n';
    if(single == 0)
        return 0;
    single = -1;
    while(!party[++single]);
    print_id(single, false);
    for(single++; single<100000; single++)
        if(party[single])
            print_id(single, true);
}

PS.
要注意没有单身狗的情况。