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

# 백준 1181번 URL

1181번: 단어 정렬

 

1. 길이가 짧은 것

2. 길이가 같을 경우 사전 순으로

 

입력할 문자열의 수 N을 먼저 입력하고, N가지의 문자열을 입력한다.

그리고 각 N개의 문자열을 위의 두 가지 조건을 만족시켜 출력하는 문제였다.

 

문제 풀이는 간단하다. C++ STL에서 지원하는 sort를 사용하는 대신 세번째 인수로 bool 형식의 compare라는 함수를 사용하면 된다.

compare라는 함수는 길이가 string.length() 함수를 사용하여 길이가 짧은 것에 우선순위를 주고 길이가 같을 경우 사전 순으로 정렬되도록 하였다.

 

#include <iostream>
#include <algorithm>
#include <string>

using namespace std;

int n;
string a[20000];

bool compare(string a, string b) {
	if (a.length() < b.length()) {
		return 1;
	}
	else if (b.length() < a.length()) {
		return 0;
	}
	else {
		return a < b;
	}
}

int main() {
	cin >> n;

	for (int i = 0; i < n; i++) {
		cin >> a[i];
	}
	sort(a, a + n, compare);

	for (int i = 0; i < n; i++) {
		if (i > 0 && a[i] == a[i - 1]) continue;
		cout << a[i] << ' ';
	}

}