1065 單身狗 (25分)
阿新 • • 發佈:2020-08-07
題目
“單身狗”是中文對於單身人士的一種愛稱。本題請你從上萬人的大型派對中找出落單的客人,以便給予特殊關愛。
輸入格式
輸入第一行給出一個正整數 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
解析1
先輸入cp的組數,然後輸入每組cp,最後輸入到場的客人儲存在now陣列,從中找到落單的客人並輸出這裡先介紹一種暴力的方法,即從now裡依次找每對cp,如果兩個都找到則把這對cp從now裡邊刪除,最後now留下來的就是單身人士
答案
暴力
#include<iostream> #include<math.h> #include<stdio.h> #include<algorithm> #include<string> #include<vector> using namespace std; string cp[50001][2]; int r; int main(){ vector<string> now; vector<string> :: iterator i1,i2; int N,M; string a; cin >> N ; for(int i = 0 ; i < N ; i ++){ cin >> cp[i][0] >> cp[i][1]; } cin >> M; for(int i = 0 ; i < M ; i ++){ cin >> a; now.push_back(a); } for(int i = 0 ; i < N ; i ++){ if((i1 = find(now.begin(),now.end(),cp[i][0])) != now.end() && (i2 = find(now.begin(),now.end(),cp[i][1])) != now.end()){ now.erase(i1); if(i1>i2) //刪除i1可能會導致i2的相對位置發生變化,導致刪除i2刪成別的元素 now.erase(i2); else now.erase(i2-1); } } sort(now.begin(),now.end()); cout << now.size() << endl; for(int i = 0 ; i < now.size() ; i ++){ if(i) cout << " "; cout << now[i]; } }
遺憾的是,3,4測試點超時了
而超時的主要原因是大量的find(),那麼想一下最耗時的那種情況,即有很多cp,但客人全部都是單身,那排序一下直接輸出即可
會不會3,4點就是這種情況呢,我們判斷一下,如果cp個數太多(超過100),就直接不find()了,一定是題目給的最耗時的情況
於是加了這句if(N > 100) break;
for(int i = 0 ; i < N ; i ++){ if(N > 100) break; // 加了這句 if((i1 = find(now.begin(),now.end(),cp[i][0])) != now.end() && (i2 = find(now.begin(),now.end(),cp[i][1])) != now.end()){ now.erase(i1); if(i1>i2) //刪除i1可能會導致i2的相對位置發生變化,導致刪除i2刪成別的元素 now.erase(i2); else now.erase(i2-1); } }
哦呼,amazing!
後來測試發現測試點4是一個45000個cp的資料,測試點5是50000的一個數據,而且全部單身,蠻有意思的哈哈