1065 單身狗 (25 分)
- 2019 年 10 月 30 日
- 筆記
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> #include <vector> #include <set> using namespace std; int main(){ int N; scanf("%d",&N); vector<int> couple(100000, -1);//初始化couple,全部賦值為-1; int a,b; for(int i = 0; i < N; i++){ scanf("%d %d", &a, &b); couple[a] = b; couple[b] = a; } int M; scanf("%d", &M); int c[M]; int count = M; vector<int> isExist(100000); for(int i = 0; i < M; i++){ scanf("%d", &c[i]); if(couple[c[i]] != -1){ isExist[couple[c[i]]] = 1; } } set<int> res; for(int i = 0; i < M;i++){ if(isExist[c[i]] != 1){ res.insert(c[i]); } } printf("%dn",res.size()); for(set<int>::iterator it = res.begin(); it != res.end();it++){ if (it != res.begin()) printf(" "); printf("%05d",*it); } return 0; }
【思路】
- 對於伴侶的存儲,使用了vector,並且各作為對方的索引進行存儲。
- 對於判斷是否是單身,使用了isExist向量,如果發現並不是單身,則在isExist中對應的索引位置存1,表示他是有對象的。
- 使用set來存儲最終結果,遍歷與會人員,若發現isExist不為1,說明沒有對象到場。
- 輸出即可。