[Baekjoon] 1181번: 단어 정렬

🥅 문제

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

 

🔍 설계

1차로 단어 길이 순으로 정렬, 2차로 사전 순으로 정렬하고, 중복을 제거하면 된다.

 

중복을 제거하기 위해 unique와 erase를 사용한다. unique는 중복이 있는 원소들을 벡터의 끝으로 보내고, 중복 원소들의 시작점을 반환한다. erase는 erase(iterator first, iterator last); 에서 [first, last) 범위의 원소를 제거한다.

 

중복을 제거했으므로, 남아있는 원소의 개수는 n이 아닐 수 있으므로 주의한다.

 

👊 풀이

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

int n;
string s;
vector<string> words;
bool compare(string a, string b) {
	if (a.length() < b.length()) {
		return true;
	}
	else if (a.length() == b.length()) {
		return a < b;
	}
	else {
		return false;
	}
}

int main(void)
{
	ios_base::sync_with_stdio(false);
	cin.tie(nullptr);
	cin >> n;
	for (int i = 0; i < n; i++) {
		cin >> s;
		words.push_back(s);
	}
	sort(words.begin(), words.end(), compare);
	words.erase(unique(words.begin(), words.end()), words.end());
	for (int i = 0; i < words.size(); i++) {
		cout << words[i] << "\n";
	}
	return 0;
}