/ SeriousOJ /

Record Detail

Wrong Answer


  
# Status Time Cost Memory Cost
#1 Accepted 1ms 532.0 KiB
#2 Accepted 2ms 640.0 KiB
#3 Accepted 2ms 844.0 KiB
#4 Wrong Answer 2ms 576.0 KiB
#5 Accepted 4ms 780.0 KiB
#6 Accepted 5ms 2.445 MiB
#7 Wrong Answer 5ms 2.816 MiB

Code

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

string solve(int n, int k, vector<string>& arr) {

    priority_queue<pair<string, int>> pq;

    for (int i = 0; i < n - 1; ++i) {
        pq.push({arr[i] + arr[i + 1], i});
    }
    for (int op = 0; op < k && !pq.empty(); ++op) {
     
        auto top = pq.top();
        pq.pop();
        string max_concat = top.first;
        int pos = top.second;

        arr[pos] = max_concat;
        arr.erase(arr.begin() + pos + 1);
        --n;

    
        if (pos > 0) {
            pq.push({arr[pos - 1] + arr[pos], pos - 1});
        }
        if (pos < arr.size() - 1) {
            pq.push({arr[pos] + arr[pos + 1], pos});
        }
    }

    return *max_element(arr.begin(), arr.end());
}

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);

    int t;
    cin >> t;

    while (t--) {
        int n, k;
        cin >> n >> k;
        vector<string> str(n);

        for (int i = 0; i < n; ++i) {
            cin >> str[i];
        }

        cout << solve(n, k, str) << endl;
    }

    return 0;
}

Information

Submit By
Type
Submission
Problem
P1083 Number concatenation
Contest
Bangladesh 2.0
Language
C++20 (G++ 13.2.0)
Submit At
2024-08-16 17:40:04
Judged At
2024-11-11 03:11:10
Judged By
Score
25
Total Time
5ms
Peak Memory
2.816 MiB