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

728x90
반응형

뒤에서부터 한 문자씩 끊어서 vector에 저장한 다음 사전순 정렬했다

 

#define _CRT_SECURE_NO_DEPRECATE
#include <iostream>
#include <string>
#include <vector>
#include <algorithm>
using namespace std;

int t,k;

int main(void) {

	ios::sync_with_stdio(false);
	cin.tie(NULL);

	cin >> t;
	string s;

	for (int tc = 1; tc <= t; tc++) {
		cin >> k;
		cin >> s;
		vector<string> list;
		string tmp = "";
		for (int i = s.length()-1; i>=0; i--) {
			tmp = s[i]+tmp;
			list.push_back(tmp);
		}

		sort(list.begin(), list.end());

		cout << "#" << tc << " " << list[k-1] << "\n";
	}


	return 0;
}
728x90
반응형
TAGS.

Comments