PTA 1121 Damn Single

huang balloon

于 2021-03-12 14:55:28 发布

阅读量109

点赞数

分类专栏： PAT 文章标签：算法

本文链接：https://blog.youkuaiyun.com/qaqaqa666/article/details/114695890

版权

PAT 专栏收录该内容

87 篇文章

订阅专栏

本文介绍了一种使用 C++ 实现的单身匹配算法。通过利用 map 和 set 数据结构，该算法可以高效地处理个体及其配偶之间的关系，对于无配偶的个体进行标记。本文详细解释了算法流程，并提供完整的代码实现。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述

在这里插入图片描述

分析：利用map存储配偶，利用set存储单身人数，假设此人没有配偶则直接插入set，假设此人有配偶，判断set中是否已有该人的配偶，没有的话插入此人，有的话删除此人的配偶。

#include<unordered_map>
#include<set>
#include<algorithm>
#include<cstdio>
using namespace std;
typedef long long ll;

unordered_map<ll,ll> um;
set<ll> s;

int main(){
    ll x,y;
    int n;
    scanf("%d",&n);
    while(n--){
        scanf("%lld %lld",&x,&y);
        um[x]=y;
        um[y]=x;
    }
    scanf("%d",&n);
    while (n--)
    {
        scanf("%lld",&x);
        if(!um.count(x)){
            s.insert(x);
        }
        else if(!s.count(um[x])){
                 s.insert(x);
        }
        else{
            s.erase(um[x]);
        }
    }
    printf("%d\n",s.size());
    for(auto i=s.begin();i!=s.end();i++){
        if(i==s.begin()){
           printf("%05lld",*i);
        }
        else{
           printf(" %05lld",*i);
        }
    }
}