結果

問題 No.2740 Old Maid
ユーザー kekenx
提出日時 2024-05-26 16:02:34
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 546 ms / 2,000 ms
コード長 1,020 bytes
コンパイル時間 1,147 ms
コンパイル使用メモリ 84,512 KB
最終ジャッジ日時 2025-02-21 16:46:53
ジャッジサーバーID
（参考情報） 		judge1 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 62
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <cassert>
#include <vector>
#include <map>

using namespace std;


int main() {
  int N; cin >> N;
  vector<int> p(N);
  map<int, int> mp1, mp2;
  for (int i = 0; i < N; i++) {
    cin >> p[i];
    mp1[p[i]] = i;
    mp2[i] = p[i];
  }

  vector<int> q;
  while (mp1.size() > 0) {
    if (mp1.size() == 2) {
      q.push_back(mp2.begin()->second);
      q.push_back(mp2.rbegin()->second);
      break;
    }
    auto [last_idx, last_value] = *mp2.rbegin();
    auto fi = mp1.begin();
    if (fi->second == last_idx) {
      fi++;
    }

    auto fi_itr = mp2.find(fi->second);
    auto se_itr = next(fi_itr);

    int fi_idx = fi_itr->first;
    int se_idx = se_itr->first;
    int fi_val = fi_itr->second;
    int se_val = se_itr->second;
    q.push_back(fi_val);
    q.push_back(se_val);

    mp2.erase(fi_idx);
    mp2.erase(se_idx);
    
    mp1.erase(fi_val);
    mp1.erase(se_val);
  }

  for (int i = 0; i < N; i++) {
    cout << q[i] << (i + 1 == N? '\n': ' ');
  }
  return 0;
}
0