[c++] 백준 1181 단어 정렬

백준 단계별로 풀어보기 [정렬] 단어 정렬

https://www.acmicpc.net/problem/1181

1181번: 단어 정렬

 

[풀이]

string을 저장하는 배열을 만들고, 길이가 짧은 순으로, 만약 길이가 같다면 사전 순으로 정렬한다. cmp 함수에 이와 같은 조건을 정해주고 sort 함수 세번째 인자로 전달해준다. 정렬 후 같은 단어가 있다면 한 번만 출력하도록 continue를 걸어준다.  

 

[코드]

#include <iostream>
#include <algorithm>

bool cmp(const std::string& a, const std::string& b) {
	//길이가 짧은 것부터, 같으면 사전 순으로.
	if (a.length() == b.length()) {
		return a < b;
	}
	else return a.length() < b.length();
}

int main() {
	int n;
	std::cin >> n;
	std::string s;
	std::string* arr = new std::string[n];
	for (int i = 0; i < n; i++) {
		std::cin >> arr[i];
	}

	sort(arr, arr + n, cmp);
	
	for (int i = 0; i < n; i++) {
		if (arr[i] == arr[i + 1])
			continue;
		std::cout << arr[i] << "\n";
	}

	return 0;
}