1121 Damn Single

1121 Damn Single

题目大意

给出n个参加派对人士的编号，请找出他们中还是单身或者伴侣没来的人士，以便特殊照顾~

核心思路

使用map<int,int> mp;存储所有伴侣的编号(双向)，使用set s;存储所有参加派对人的编号。使用set s2;存放单身和伴侣没有来的人士。
遍历一遍参加派对的人的编号，如果发现此人没有伴侣或者伴侣没来参加，则把此人的编号放入集合s2。

代码

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <map>
#include <set>
using namespace std;
map<int,int> mp;//存放所有键值对
set<int> s;//存放所有参加聚会的人
int main(){
    //把所有的键值对都放到mp中去
    int n;
    cin>>n;
    while(n--){
        int ans1,ans2;
        cin>>ans1>>ans2;
        mp[ans1]=ans2;
        mp[ans2]=ans1;
    }
    //把所有参加派对的人放到集合s中
    cin>>n;
    while(n--){
        int ans;
        cin>>ans;
        s.insert(ans);
    }
    //遍历集合s中的所有元素，看看这个人是不是单身的，记录单身人士的编号
    set<int> s2;//存放单身和伴侣没有来的人士
    for(auto it=s.begin();it!=s.end();it++){
        if(mp.find(*it)==mp.end()||s.find(mp[*it])==s.end()){//此人没有伴侣或者伴侣没来参加
            s2.insert(*it);
        }
    }
    //输出
    cout<<s2.size()<<endl;
    for(auto it=s2.begin();it!=s2.end();it++){
        if(it!=s2.begin()) cout<<" ";
        printf("%05d",*it);
    }
}

注意点

1. 如果容器找不到某个元素，都会返回它的尾元素后一元素的迭代器。