Algorithm
1256. [S/W 문제해결 응용] 6일차 - K번째 접미어
이무쿤
2020. 5. 5. 10:47
반응형
SW Expert Academy
SW 프로그래밍 역량 강화에 도움이 되는 다양한 학습 컨텐츠를 확인하세요!
swexpertacademy.com
문제 조건대로 접미어 뽑아서 정렬후 인덱스 출력
#include<iostream>
#include<string>
#include<vector>
#include<algorithm>
using namespace std;
int K;
string str;
vector<string> vec;
int main(int argc, char** argv)
{
int test_case;
int T;
cin >> T;
/*
여러 개의 테스트 케이스가 주어지므로, 각각을 처리합니다.
*/
for (test_case = 1; test_case <= T; ++test_case)
{
cin >> K >> str;
vec.clear();
for (int i = 0; i < str.size(); i++)
{
string temp;
for (int j = i; j < str.size(); j++)
{
temp += str[j];
}
vec.push_back(temp);
}
sort(vec.begin(), vec.end());
cout << "#" << test_case << " " << vec[K - 1] << endl;
}
return 0;//정상종료시 반드시 0을 리턴해야합니다.
}
반응형