K번째 문자열

https://swexpertacademy.com/main/code/problem/problemDetail.do?contestProbId=AV18KWf6ItECFAZN&categoryId=AV18KWf6ItECFAZN&categoryType=CODE&problemTitle=K%EB%B2%88%EC%A7%B8&orderBy=FIRST_REG_DATETIME&selectCodeLang=ALL&select-1=&pageSize=10&pageIndex=1

SW Expert Academy

 

부분 문자열을 구하는 것은

투 포인터로 하고,

중복과 정렬은 set로 구현.

 

set는 index로 접근을 할 수 없다.

iterator를 사용하자.

 

<소스 코드>

#include <iostream>
#include <string>
#include <set>
 
using namespace std;
 
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);
 
    int test;
    cin >> test;
 
    for (int t = 1; t <= test; t++)
    {
        int k;
        string s;
        cin >> k >> s;
 
        // 중복 제거, 정렬
        set<string> ans;
 
        // 투 포인터
        for (int i = 0; i < s.length(); i++)
        {
            for (int j = 0; j < s.length(); j++)
                ans.insert(s.substr(i, j + 1));
        }
 
        cout << '#' << t << ' ';
 
        if (ans.size() < k)
            cout << "none" << '\n';
        else
        {
            int cnt = 0;
            set<string>::iterator iter = ans.begin();
            while (cnt != k - 1)
            {
                iter++;
                cnt++;
            }
            cout << *iter << '\n';
        }
    }
    return 0;
}