PTA乙级 （1065 单身狗 (25分)（map,set.find(),vector））

1065 单身狗 (25分)

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

解题思路：一开始只使用map<int, int> comp和vector<int> vec;comp用来存储每一对数字，vec用来存储查找出来的单身狗，查找单身狗时用了双重for循环直接一个个用map查询是否配对结果测试点三，四超时；后来改成使用set.find()查找：

1.将所有情侣用一个set存起来,一对情侣再分别做两个映射,方便查询11111-->22222,22222-->11111
2.客人用一个数组存起来,再用set存一遍方便查询,接下来就遍历客人,判断是否落单
3.set是一颗二叉树,这里使用了set的find函数

#include <iostream>
#include <algorithm>
#include <cmath>
#include <cstring>
#include <string>
#include <cstdio>
#include <map>
#include <vector>
#include <set>
using namespace std;
int main()
{
    ios::sync_with_stdio(false);   
    cin.tie(0);
    map<int, int> comp;
    set<int> st1,st2;
    vector<int> vec;
    int n,m,temp1,temp2;
    int s1[10002];
    cin>>n;
    for(int i=0;i<n;i++)
    {
        cin>>temp1>>temp2;
        st1.insert(temp1);
        st1.insert(temp2);
        comp[temp1]=temp2;
        comp[temp2]=temp1;
    }
    cin>>m;
    for(int i=0;i<m;i++) 
    {cin>>s1[i];st2.insert(s1[i]);}
    for(int i=0;i<m;i++)
    {
        if(st1.find(s1[i])!=st1.end())
        {
            if(st2.find(comp[s1[i]])!=st2.end());
            else vec.push_back(s1[i]);
        }
        else vec.push_back(s1[i]);
    }
    sort(vec.begin(),vec.end());
    cout<<vec.size()<<endl;
    for(int i=0;i<vec.size();i++)
    {
        if(i==0) printf("%05d",vec[i]);
        else printf(" %05d",vec[i]);
    }
    return 0;
}