C++/Python PAT乙级1065 单身狗 (25分)

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

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

#include <iostream>
#include <algorithm>
using namespace std;
bool cmp(int a,int b)
{
    return a>b;
}
int main()
{
    int n;
    cin >> n;
    int couple[100000]={-1};
    int flags[100000]={0};
    for (int i=0;i<n;i++)
    {
        int a,b;
        cin >> a >> b;
        couple[a]=b;//这里是重点，后续方便查找，了解了后面就简单了
        couple[b]=a;
    }
    int k;
    cin >> k;
    int id[100000]={-1};
    int p=0;
    for(int i=0;i<k;i++)
    {
        int a;
        cin >> a;
        id[a]=a;
        p++;
        if(couple[a]!=-1) {flags[a]++;flags[couple[a]]++;}
        if(flags[a]==2 && flags[couple[a]]==2) {id[a]=-1;id[couple[a]]=-1;p-=2;}
    }
    if(p==0) cout << 0;
    //cout << p;
    else if(p>0) 
    {
        cout<<p<<endl;
        sort(id,id+100000,cmp);
        for(int i=p-1;i>=0;i--)
        {
            printf("%05d",id[i]);
            if(i>0) cout << ' ';
        }
    }
    return 0;
}

Python

测试点3超时

n=int(input())
d1,d2={},{}
for i in range(n):
    l1,l2=input().split()
    d1[l1]=l2
    d2[l2]=l1
m=int(input())
lst=list(input().split())
ans=[]
for i in lst:
    Flag=True
    if (i in d1 and d1[i] in lst) or (i in d2 and d2[i] in lst):
        Flag=False
    if Flag:
        ans.append(i)
print(len(ans))
if len(ans)!=0:
    ans.sort()
    print(" ".join(ans))