[C++][Goorm][정렬] 단어장 만들기

구름EDU - 모두를 위한 맞춤형 IT교육

 

*해결법

처음엔 단어 길이로 오름차순 정렬되는 것을 못보고 그냥 sort 했다가 틀렸다.

문제에서 요구하는 것은 단어의 길이가 같을 땐 알파벳 순서로 정렬하고 단어의 길이가 다를 땐 단어의 길이별로 정렬해야 하므로 Custom Sort 함수가 필요하다.

 

*알아둘것

1) Custom Sort 함수 활용법

bool sortWord(string word1, string word2){
	//사이즈가 다르면 긴게 뒤로
	if(word1.size() != word2.size()){
		return word1.size() < word2.size();
	}
	//사이즈가 같으면 큰 값이 뒤로
	return word1 < word2;
}

sort(words.begin(), words.end(), sortWord);

참값인게 뒤로 가므로 return bool 시 위와 같이 적어주어야 한다.

 

2) string 길이구하는 법

str.size();

3) string 크기?값 비교

str1 < str2

 

*코드

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

bool sortWord(string word1, string word2){
	//사이즈가 다르면 긴게 뒤로
	if(word1.size() != word2.size()){
		return word1.size() < word2.size();
	}
	//사이즈가 같으면 큰 값이 뒤로
	return word1 < word2;
}

int main() {
	int wordNum = 0, wordIndex = 0;
	vector<string> words(wordNum);
	string word;
	
	cin >> wordNum >> wordIndex;
	for(int i = 0; i<wordNum; i++){
		cin >> word;
		words.push_back(word);
	}
	
	sort(words.begin(), words.end(), sortWord);

	cout << words[wordIndex-1] << endl;
	return 0;
}

 

 

* vector pair의 second 값을 기준으로 정렬하기 위한 custom sort 는 아래 게시글 참고

[BOJ][C++] 1931 회의실 배정 :: 라인 스위핑