[BOJ/C++] 단어 정렬(1181번)

1. 개요

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

1181번: 단어 정렬

1-1. 설명

알파벳 소문자로 이루어진 N개의 단어가 들어오면 아래와 같은 조건에 따라 정렬하는 프로그램을 작성한다.

 1. 길이가 짧은 것

 2. 길이가 같으면 사전순

1-2. 제한 사항

 - 첫 줄에 단어의 개수 N이 주어지며,  N은 1 이상 20000이하 자연수

 - 둘째 줄부터 N개 줄에 걸쳐 알파벳 소문자로 이루어진 단어가 한줄에 하나씩 주어지며, 문자열 길이는 50을 넘지 않음

---

2. 구현

2-1. 풀이

sort 함수를 이용해 정렬 조건을 지정하여 정렬하였다. 중복은 출력할 때 제거해주었다. 

2-2. 코드

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

constexpr size_t MAX_NUM = 20000;
string words[MAX_NUM];
bool compare(string a, string b);

int main()
{
	ios_base::sync_with_stdio(0);
	cin.tie(0);
	cout.tie(0);
	int N;
	string preStr;

	cin >> N;
	for (int i = 0; i < N; i++) {
		cin >> words[i];
	}

	sort(words, words + N, compare);

	for (int i = 0; i < N; i++) {
		if (i == 0) preStr = words[i];
		else if (preStr == words[i]) continue;
		preStr = words[i];
		cout << words[i] << "\n";
	}
	return 0;
}

bool compare(string a, string b) {
	if (a.size() < b.size()) return true;
	else if (a.size() == b.size()) return a < b;
	else return false;
}