Yukicoder No.205 マージして辞書順最小

solution

基本的にはpriority_queueに突っ込んで小さい方から先頭を削っていけばよい。$O(\Sigma_N|S_i|)$。 ただし比較の際に一方が他方のprefixになっている際は特殊化が必要で、例えばzaはzより小さくならねばならない。 これは単純に書き下すこともできるが、各文字列の末尾に入力のどの文字よりも大きい文字を番兵のように追加しておくことでも実装できる。

implementation

#include <iostream>
#include <queue>
#define repeat(i,n) for (int i = 0; (i) < (n); ++(i))
using namespace std;
template <class T> using reversed_priority_queue = priority_queue<T, vector<T>, greater<T> >;
int main() {
    int n; cin >> n;
    reversed_priority_queue<string> que;
    repeat (i,n) {
        string s; cin >> s;
        s += '\xff';
        que.push(s);
    }
    string ans = "";
    while (not que.empty()) {
        string t = que.top(); que.pop();
        ans += t[0];
        if (t.size() != 2) { // 1 is for the sentinel
            que.push(t.substr(1));
        }
    }
    cout << ans << endl;
    return 0;
}