[ SWEA 1256 ] K번째 접미어 (C++)

SWExpertAcademy의 K번째 접미어(1256 / D5) 문제이다.

[ 문제풀이 ]

1) 주어진 문자열에서 찾을 수 있는 모든 접미어들 중에서 K번째 오는 접미어를 찾아야 하는 문제이다.

   본인은, 입력을 받은 후에 접미어들을 string 배열에 모두 저장해 주었고, 이 배열을 정렬 시켜주었다.

   그 후, K가 주어진 접미어의 갯수보다 크다면 none을 그게 아니라면 K번 인덱스에 매칭되어 있는 접미어를

   출력시켜 주었다.

[ 소스코드 ]

#include<iostream>
#include<string>
#include<algorithm>
 
#define MAX 400
#define endl "\n"
using namespace std;
 
int N, Idx;
string Arr[MAX];
string Answer;
 
void Initialize() {}
 
void Input()
{
    cin >> N;
    N--;
    string S; cin >> S;
    Idx = 0;
    for (int i = 0; i < S.length(); i++)
    {
        string Temp = "";
        for (int j = i; j < S.length(); j++)
        {
            Temp = Temp + S[j];
        }
        Arr[Idx++] = Temp;
    }
}
 
void Solution()
{
    sort(Arr, Arr + Idx);
    if (N >= Idx) Answer = "none";
    else Answer = Arr[N];
}
 
void Solve()
{
    int Tc; cin >> Tc;
    for (int T = 1; T <= Tc; T++)
    {
        Initialize();
        Input();
        Solution();
 
        cout << "#" << T << " " << Answer << endl;
    }
}
 
int main(void)
{
    ios::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);
 
    //freopen("Input.txt", "r", stdin);
    Solve();
 
    return 0;
}