1065. 单身狗(25)

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

输入格式:

输入第一行给出一个正整数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

解题思路:创建两个大数组a和b,a用来存储夫妻关系,b用来表示对应id的人是否出席。0表示没有出席,1表示出席。对于a数组,以自己id作为下标,配偶id作为数组项内容存储。则在进行遍历是否是单身狗时,可以通过a数组得到的配偶id作为b数组的下标,来判断是否为1,判断是否出席。若出席则不是单身狗。

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

int a[100000]={},b[100000]={};
vector<int> v;

int main() {
    int m,n,index,i=0,flag=0;
    cin>>m;
    while(m--){
        cin>>index;
        cin>>a[index];
        a[a[index]]=index;
    }
    cin>>n;
    while(n--){
        cin>>index;
        v.push_back(index);
        if(b[a[index]]==1){
            b[a[index]]=0;
            i--;}
        else {
            b[index]=1;
            i++;}
    }
    cout<<i<<endl;
    sort(v.begin(),v.end());
    for(vector<int>::iterator iter=v.begin();iter!=v.end();iter++){
        if(b[*iter]==1)
            printf("%s%05d",flag==0?"":" ",*iter);
            flag=1;}
}

猜你喜欢

转载自blog.csdn.net/qq_34228570/article/details/79919323