7-12 单身狗 (25 分)(set用法)

“单身狗”是中文对于单身人士的一种爱称。本题请你从上万人的大型派对中找出落单的客人,以便给予特殊关爱。

输入格式:

输入第一行给出一个正整数 N(≤50000),是已知夫妻/伴侣的对数;随后 N 行,每行给出一对夫妻/伴侣——为方便起见,每人对应一个 ID 号,为 5 位数字(从 00000 到 99999),ID 间以空格分隔;之后给出一个正整数 M(≤10000),为参加派对的总人数;随后一行给出这 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

题解:先将对应的lover标记,然后循环m个人看其lover是否在里面,还有如果这个人就没有被标记,就将这些人放入set,输出即可。

#include
using namespace std;
const int maxn=100050;
sets;
int a[maxn],b[maxn],c[maxn];
int main()
{
    memset(a,-1,sizeof(a));//初始化为-1代表都是单身
    int n;
    cin>>n;
    for(int i=1;i<=n;i++)
    {
        int x,y;
        cin>>x>>y;
        a[x]=y;//分别记录lover
        a[y]=x;
    }
    int m;
    cin>>m;
    for(int i=1;i<=m;i++)
    {
        cin>>b[i];
        if(a[b[i]]!=-1)//将不是单身的标记
        {
            c[a[b[i]]]=1;
        }
    }
    for(int i=1;i<=m;i++)//注意上面循环完,才能判断是不是单身
    {
        if(!c[b[i]])//如果该人是单身,就将其放入set
            s.insert(b[i]);
    }
    cout<::iterator it;
    for(it=s.begin();it!=s.end();it++)
    {
        if(it==s.begin())
            cout<

 

你可能感兴趣的:(STL,排序)