B1065 单身狗

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

输入格式:

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

最终AC的代码如下:
#include <iostream>
#include <string>
#include <vector>
#include <set>
#include <map>
using namespace std;

int main(){
    int i, n, m, len;
    string t1, t2;
    vector<string> ans;
    map<string, string> mp;
    map<string, string>::iterator mit;
    set<string> si;
    set<string>::iterator sit;
    scanf("%d", &n);
    for(i=0; i<n; i++){
        cin>>t1;
        getchar();
        cin>>t2;
        mp[t1] = t2;
        mp[t2] = t1;
    }
    scanf("%d", &m);
    for(i=0; i<m; i++){
        cin>>t1;
        getchar();
        si.insert(t1);
    }
    for(set<string>::iterator it=si.begin(); it!=si.end(); it++){
        mit = mp.find(*it);
        if(mit!=mp.end()){
            sit = si.find(mit->second);
            if(sit==si.end()){
                ans.push_back(*it);
            }
        }else{
            ans.push_back(*it);
        }
    }
    len = ans.size();
    printf("%d\n", len);
    for(i=0; i<len; i++){
        if(i>0){
            printf(" ");
        }
        cout<<ans[i];
    }
    if(len>0){
        printf("\n");
    }
    return 0;
}

提示:这里需要注意,len可能为0,因此输出第二个换行前,需先判断一下,否则有一个测试用例将提示格式错误。

总的思路是,巧用map和set这里两个工具,由于二者都是有序的,所以查找的时间很快。

记录此题的目的是,貌似判断len>0与判断len!=0的速度要更快些,尤其是当比较次数比较大时,这种情况更加明显。

猜你喜欢

转载自www.cnblogs.com/heyour/p/12238433.html