1. 程式人生 > 實用技巧 >1065 單身狗 (25分)

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

解析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的一個數據,而且全部單身,蠻有意思的哈哈