PAT乙级-1065 单身狗(25 分)

题目链接

#include<iostream>
#include<vector>
#include<set>
using namespace std;

int main(){
    int n,m;
    cin>>n;
    vector<int> couple(100000,-1);
    int a,b;
    for(int i=0;i<n;i++){
        cin>>a>>b;
        couple[a]=b;
        couple[b]=a;
    }
    vector<int> guest(m),partner(100000);
    cin>>m;
    for(int i=0;i<m;i++){
        cin>>guest[i];
        if(couple[guest[i]]!=-1) //如果有对象
            partner[couple[guest[i]]]=1; //将对象标记为存在
    }
    set<int> st;
    for(int i=0;i<m;i++){
        if(!partner[guest[i]]) st.insert(guest[i]); //没有发现他的对象，则插入
    }
    cout<<st.size()<<endl;
    for(auto it=st.begin();it!=st.end();it++){
        if(it!=st.begin()) cout<<" ";
        printf("%05d",*it); //注意输出格式
    }
    return 0;
}