Submission #2837731

---

Source Code Expand

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

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

    string s;
    int k;
    cin >> s >> k;

    vector<string> vec;
    vec.push_back(s.substr(0, 1));

    unsigned long max_vec_size = s.size() < 5 ? s.size() : 5;
    for (int i = 0; i < s.length(); i++) {
        unsigned long max_sub_length = min(s.length() - i, max_vec_size);
        for (int j = 1; j <= max_sub_length; ++j) {
            string sub_s = s.substr(i, j);
            if (sub_s == "") break;
            unsigned long vec_length = vec.size();
            for (int n = 0; n < vec_length; n++) {
                if (vec[n] == sub_s) {
                    break;
                } else if (vec[n] > sub_s) {
                    vec.insert(vec.begin() + n, sub_s);
                    if (vec.size() > 5) vec.pop_back();
                    break;
                } else if (n == vec_length - 1 && vec_length < 5 ) {
                    vec.push_back(sub_s);
                }
            }
        }
    }

    cout << vec[k - 1] << '\n';

    return 0;
}

Submission Info

Judge Result