PAT 乙级 1065 单身狗 (25分)

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

输入第一行给出一个正整数 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
作者: CHEN, Yue
单位: 浙江大学
时间限制: 200 ms
内存限制: 64 MB
代码长度限制: 16 KB

这题写起来很容易写，但是容易超时。为了优化时间效率，可以在第一次读入参加派对人的时候将单身的加入字典中，不是单身的加入一个vector中，然后在对vector中的数据进行遍历，可以减少循环的次数

#include <iostream>
#include <set>
#include <vector>
#include <iterator>
#include <algorithm>
using namespace std;
int main(){
    int n,m,mid_1,mid_2,store[100005];
    set<int> ans;
    vector<int> vec;
    cin>>n;
    for (int i = 0; i < 100005; ++i) {
        store[i] = -1;
    }
    for (int i = 0; i < n; ++i) {
        scanf("%d %d",&mid_1,&mid_2);
        store[mid_1] = mid_2;
        store[mid_2] = mid_1;
    }
    cin>>m;
    for (int i = 0; i < m; ++i) {
        cin>>mid_1;
        if(store[mid_1] == -1)
            ans.insert(mid_1);
        else
            vec.push_back(mid_1);
    }
    for (vector<int>::iterator it = vec.begin(); it != vec.end(); it++) {
        if(find(vec.begin(), vec.end(), store[*it]) == vec.end())
            ans.insert(*it);
    }
    cout<<ans.size()<<endl;
    for (set<int>::iterator it = ans.begin(); it != ans.end(); it++) {
        if(it == ans.begin())
            printf("%05d",*it);
        else
            printf(" %05d",*it);
    }
    return 0;
}