1065 单身狗(25 分)
“单身狗”是中文对于单身人士的一种爱称。本题请你从上万人的大型派对中找出落单的客人,以便给予特殊关爱。
输入格式:
输入第一行给出一个正整数 N(≤ 50 000),是已知夫妻/伴侣的对数;随后 N 行,每行给出一对夫妻/伴侣——为方便起见,每人对应一个 ID 号,为 5 位数字(从 00000 到 99999),ID 间以空格分隔;之后给出一个正整数 M(≤ 10 000),为参加派对的总人数;随后一行给出这 M 位客人的 ID,以空格分隔。题目保证无人重婚或脚踩两条船。
输出格式:
首先第一行输出落单客人的总人数;随后第二行按 ID 递增顺序列出落单的客人。ID 间用 1 个空格分隔,行的首尾不得有多余空格。
输入样例:
13 211111 22222 333333 44444 455555 66666 57 655555 44444 10000 88888 22222 11111 23333
输出样例:
15 210000 23333 44444 55555 88888
PS:看到这题目,当场去世!!!!
我的思路:首先录入伴侣的信息,然后当输入参加派对的那些人时,把他的key值置1,然后遍历一遍所有人,当满足单身狗条件时(本身key为1、伴侣的key为0),记录这组数据。随后输出就行。(显然我这里最后存储的时候用map就有点大材小用了,其实用set就行了,懒得加头文件和定义了)
注意:1、考虑当count为0的情况,否则测试点2会出现段错误,运行超时;
2、注意时间复杂度要低,否则测试点3、4运行超时。
1#include<iostream> 2#include<map> 3#include<string> 4using namespace std; 5int main(){ 6 map<string, pair<string,int>> s, res; //s用于判断是否单身 7 map<string, pair<string,int>>::iterator it; 8 int n, m, count = 0; 9 string r, l; 10 cin >> n; 11 while (n--) { 12 cin >> l >> r; 13 s[l].first = r; //存入伴侣信息 14 s[r].first = l; 15 } 16 cin >> m; 17 for (int i = 0; i < m; i++) { 18 cin >> l; 19 s[l].second++; 20 } 21 for (it = s.begin(); it != s.end(); it++) { 22 if (it->second.second && !s[it->second.first].second) { //逻辑判断:自己出席了会议,伴侣没有 23 count++; //记录单身狗数目 24 res[it->first].second++; //将符合条件的人存到res中 25 } 26 } 27 cout << count << endl; 28 if (count) { //注意count为0时,就不用输出ID了,否则测试点2出现段错误、运行超时 29 it = res.begin(); 30 cout << it++->first; 31 for (; it != res.end(); it++) { 32 cout << " " << it->first; 33 } 34 } 35 return 0; 36}