結果

問題 No.775 tatyamと素数大富豪(hard)
ユーザー ei1333333ei1333333
提出日時 2018-12-22 00:59:56
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
TLE  
実行時間 -
コード長 1,449 bytes
コンパイル時間 2,807 ms
コンパイル使用メモリ 228,992 KB
実行使用メモリ 19,592 KB
最終ジャッジ日時 2024-10-14 01:27:25
合計ジャッジ時間 6,717 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
10,496 KB
testcase_01 AC 2 ms
5,248 KB
testcase_02 AC 2 ms
5,248 KB
testcase_03 AC 2 ms
5,248 KB
testcase_04 AC 2 ms
5,248 KB
testcase_05 AC 5 ms
5,248 KB
testcase_06 AC 19 ms
5,248 KB
testcase_07 AC 180 ms
9,372 KB
testcase_08 AC 181 ms
9,372 KB
testcase_09 TLE -
testcase_10 -- -
testcase_11 -- -
testcase_12 -- -
testcase_13 -- -
testcase_14 -- -
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>

using namespace std;

int N, K;

int main() {
  cin >> N >> K;
  int Z = K;
  vector< string > P(N);
  for(int i = 0; i < N; i++) {
    cin >> P[i];
  }

  vector< int > ord(N);
  iota(begin(ord), end(ord), 0);
  sort(begin(ord), end(ord), [&](const int &a, const int &b) {
    if(P[a] + P[b] == P[b] + P[a]) return P[a].size() > P[b].size();
    return P[a] + P[b] > P[b] + P[a];
  });

  string concat;
  for(auto &p : ord) concat += P[p];
  priority_queue< pair< string, vector< int > > > que;
  que.emplace(concat, ord);
  unordered_set< string > memo;
  memo.emplace(concat);

  auto normalize = [&](vector< int > v) {
    for(int i = 1; i < v.size(); i++) {
      for(int j = i; j > 0; j--) {
        if(P[v[j]] + P[v[j - 1]] == P[v[j - 1]] + P[v[j]]) {
          if(P[v[j]].size() > P[v[j - 1]].size()) {
            swap(v[j], v[j - 1]);
          }
        }
      }
    }
    return v;
  };

  while(que.size() && K > 0) {
    auto p = que.top().second;

    cout << que.top().first << endl;
    --K;

    que.pop();

    for(int i = 0; i < ord.size(); i++) {
      for(int j = i - 1; j >= 0; j--) {
        if(P[p[i]] + P[p[j]] >= P[p[j]] + P[p[i]]) continue;
        swap(p[i], p[j]);
        string con;
        for(auto &q : p) con += P[q];
        if(memo.count(con)) continue;
        memo.emplace(con);
        que.emplace(con, normalize(p));
        swap(p[i], p[j]);
        break;
      }
    }
  }
}
0