PAT Basic 1065 单身狗

1065 单身狗

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

输入格式：

输入第一行给出一个正整数 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

解题思路：

通过map建立情侣cp的对应关系 ， 再将测试的人保存到vector中, 循环检测，如果在测试中的人如果能够组成cp那么他们的就标记为0，最后排序输出即可。 最后注意测试点 1 3 我是通过数字表示的id，如果小于5位数 那么你输出的时候就应当补全printf("%05d "， v[i]),我这种做法的时间复杂度还是比较高的哈~~， 对于测试点 3 4 已经100多ms了， 但是还是很好理解的，好的方法日后在更新。

解题代码:

#include <vector>
#include <cstdio>
#include <algorithm>
#include <unordered_map>
using namespace std;
int main(){
    int cps, n, t1, t2, cnt = 0;
    scanf("%d", &cps);
    unordered_map<int, int> m;
    for(int i = 0; i < cps; i++){
        cin >> t1 >> t2;
        m[t1] = t2; m[t2] = t1;
    }
    scanf("%d", &n);
    vector<int> v(n);
    for(int i = 0; i < n; i++){
       scanf("%d", &v[i]);
    }
    for(int i = 0; i < v.size(); i++){
        int cp = m[v[i]];
        for(int j = 0; j < v.size(); j++){
            if(v[j] == 0) continue;
            if(cp == v[j]) {v[j] = 0; v[i] = 0; cnt++;}
        }
    }
    sort(v.begin(), v.end());
    int acs = v.size() - cnt * 2, vs = v.size() - 1;
    printf("%d\n", acs);
    for(int i = 0; i < vs; i++){
        if(v[i]) printf("%05d ", v[i]);
    }
    if(acs) printf("%05d", v[vs]);
    return 0;
}