1. 程式人生 > >PAT 乙級 1065. 單身狗(25)

PAT 乙級 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
#include<bits/stdc++.h>
using namespace std;

int table[100000];
int flag[100000];

int main()
{
    memset(table,-1,sizeof(table));
    memset(flag,0,sizeof(flag));
    priority_queue<int,vector<int>,greater<int> > ans;
    int n;
    cin>>n;
    for(int i=0;i<n;i++)
    {
        int a,b;
        cin>>a>>b;
        table[a]=b;
        table[b]=a;
    }
    int m;
    cin>>m;
    int in[10010];
    for(int i=0;i<m;i++)
    {
        cin>>in[i];
        flag[in[i]]=1;
    }
    for(int i=0;i<m;i++)
    {
        if(table[in[i]]!=-1)
        {
            if(flag[table[in[i]]]==0)
            {
                ans.push(in[i]);
            }
        }
        else
        {
            ans.push(in[i]);
        }
    }
    int fflag=0;
    cout<<ans.size()<<endl;
    while(ans.size()>0)
    {
        if(fflag==1)
        {
            printf(" ");
        }
        fflag=1;
        printf("%05d",ans.top());
        ans.pop();
    }
    return 0;
}