1257. [S/W 문제해결 응용] 6일차 - K번째 문자열(D6)

STL을 이용하면 쉽다.

주어진 문자열의 부분문자열은 set에 저장하면, 알아서 다 정렬이 되어있다.

#include <stdio.h>
#include <set>
#include <string>
#include <iostream>
using namespace std;
 
int main()
{
    int T;
    setbuf(stdout, NULL);
    scanf("%d", &T);
    for (int test_case = 1; test_case <= T; test_case++)
    {
        int n;
        scanf("%d", &n);
        set<string> s;
        string str;
        cin >> str;
        int len = str.size();
        int it = 0;
        for (int i = 0; i < len; i++)
        {
            for (int end = str.size()-it; 0 < end; end--)
            {
                s.insert(str.substr(it, end));
            }
            it++;
        }
        printf("#%d ", test_case);
        auto iter = s.begin();
        int check = 0;
        for (int i = 1; i < n; i++)
        {
            iter++;
            if(iter==s.end())
            { 
                check = 1;
                break;
            }
        }
        if (check)
            printf("none\n");
        else
            cout << (*iter) << '\n';
    }
    return 0;
}