단어 정렬

문제

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

해당 문제는 N개의 입력 문자열을 주어진 조건에 따라 정렬하는 문제이다.

이 문제도 이전의 "국영수" 문제와 같이 sort 함수에서 사용할 비교함수만 잘 작성하면 큰 어려움 없이 풀 수 있었다.

정렬 기준에서 가장 우선이 길이가 짧은 순서이기 때문에 string에서 제공하는 length 함수를 사용해서 문자열 간에 길이를 비교한다.
만약, 두 문자열의 길이가 같으면 사전 순서로 정렬해야 하기 때문에, 이때는 문자열 간에 대소비교를 통해 문자열을 정렬한다.

문자열을 정렬한 후 마지막으로 조심해야 하는 것은 중복되는 단어의 경우 출력하지 않도록 하는 것이다.
이를 위해 문자열을 저장한 vector에서 다음 성분의 문자열과 비교 후 같으면 출력하지 않고 다르면 출력하도록 한다.

그리고 비교함수를 작성할때 const와 참조자 선언도 잊지 않도록 해야겠다.

위의 내용을 정리한 전체 코드는 다음과 같다.

#define _CRT_SECURE_NO_WARNINGS
#include <iostream>
#include <string>
#include <algorithm>
#include <vector>
#include <stack>
#include <queue>
#include <map>
using namespace std;

bool cmp(const string& a, const string& b) {
	if (a.length() == b.length()) {
		return a < b;
	}

	return a.length() < b.length();
}

int main() {
	ios_base::sync_with_stdio(false);
	//freopen("input.txt", "rt", stdin);

	int n;
	string temp;

	cin >> n;
	vector<string> str;

	for (int i = 0; i < n; i++) {
		cin >> temp;
		str.push_back(temp);
	}

	sort(str.begin(), str.end(), cmp);

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

		cout << str[i] << "\n";
	}

	return 0;
}