PAT 1065 单身狗 (25 分) set的使用

#include <iostream>
#include <set>
#include <vector>
#include <algorithm>
using namespace std;
int main() {
   int n,temp1,temp2,m,couple[1000000]={0};
   set<int> a;
   cin>>n;
   for(int i=0;i<n;i++){
       cin>>temp1>>temp2;
       couple[temp1]=temp2;
       couple[temp2]=temp1;
    }
    cin>>m;
    vector<int> res;
    for(int i=0;i<m;i++){
        cin>>temp1;
        a.insert(temp1);
    }
    for(auto it=a.begin();it!=a.end();it++){
        if(a.find(couple[*it])==a.end()){          //p[i]的伴侣为couple[p[i]],a中找不到他的伴侣则输入res数组
            res.push_back(*it);
        }
    }
    cout<<res.size()<<endl;
    for(int i=0;i<res.size();i++){
        if(i!=0){cout<<" ";}
        printf("%05d", res[i]);
    }
    system("pause");
    return 0;
}