結果

問題 No.775 tatyamと素数大富豪(hard)
ユーザー comavius
提出日時 2023-07-11 15:44:09
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,647 bytes
コンパイル時間 2,633 ms
コンパイル使用メモリ 218,216 KB
最終ジャッジ日時 2025-02-15 09:49:05
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3 WA * 2
other AC * 8 WA * 2
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
using ll = long long;



bool adhoc_comparator(const pair<string, ll> &a, const pair<string, ll> &b) {
    ll check_len = min(a.first.size(), b.first.size());
    for (ll i = 0; i < check_len; i++) {
        if (a.first[i] != b.first[i]) {
            return a.first[i] > b.first[i];
        }
    }
    if (a.first.size() < b.first.size()){
        return adhoc_comparator(a, {b.first.substr(check_len), b.second});
    }
    if (a.first.size() > b.first.size()){
        return adhoc_comparator({a.first.substr(check_len), a.second}, b);
    }
    else {
        return a.second > b.second;
    }
}

int main() {
    ll n, k;
    cin >> n >> k;
    vector<pair<string, ll>> v;
    for (ll i = 0; i < n; i++) {
        string a;
        cin >> a;
        if (a.size() == 2){
            if (a[0] == a[1]){
                v.push_back({a.substr(0, 1), 1});
                v.push_back({a.substr(0, 1), 1});
            }
            else {
                v.push_back({a, a.size()});
            }
        }
        else {
            v.push_back({a, a.size()});
        }
    }
    set<string> ls;
    std::sort(v.begin(), v.end(), adhoc_comparator);
    priority_queue<string> pq;
    for (ll i = 0; i < k * 100; i++) {
        string s;
        for (ll j = 0; j < v.size(); j++) {
            s += v[j].first;
        }
        if (ls.find(s) == ls.end()) {
            ls.insert(s);
            pq.push(s);
        }
        next_permutation(v.begin(), v.end(), adhoc_comparator);
    }
    for (ll i = 0; i < k; i++){
        cout << pq.top() << "\n";
        pq.pop();
    }
    cout << flush;
}
0