알고리즘

BOJ 1181: 단어 정렬

minbear 2023. 5. 15. 22:39

#BOJ 1181: 단어 정렬

 

1181번: 단어 정렬

첫째 줄에 단어의 개수 N이 주어진다. (1 ≤ N ≤ 20,000) 둘째 줄부터 N개의 줄에 걸쳐 알파벳 소문자로 이루어진 단어가 한 줄에 하나씩 주어진다. 주어지는 문자열의 길이는 50을 넘지 않는다.

www.acmicpc.net

#풀이

string 으로 받아서 sort() 함수를 사용하는데 비교하는 함수를 직접 구현해서 풀었다.

그리고 또한 중복을 제거 해야해서 나는 그냥 sort 함수를 통과하면 중복된 값들은 붙어 있어 제거를 하였는데 

vector의 메서드에서 제거를 할 수 있었다.

 a.erase(unique(a.begin(), a.end()), a.end());

이렇게 사용하면 unique함수가 중복된 값을 뒤로 보내서 제거를 할 수 있다.

이 말이 이해가 안될 수도 있어서 

만약 vector a = {1,1,1,2,3}; 이라고 하면 unique 함수를 통과하면 

a = {1,2,3,1,1}

이렇게 벡터의 원소가 변하고 중복된 값의 첫번째 인덱스를 리턴한다.

그래서 erase 함수를 통해 중복을 제거 할 수 있다.

 

#코드

#include<iostream> // #include<bits/stdc++.h>
#include<algorithm>
using namespace std;
int n;
string arr[20000];
bool cmp(const string& a, const string& b) {
	if (a.size() < b.size())
		return true;
	if (a.size() == b.size()) {
		if (a < b)
			return true;
	}
	return false;
}
int main()
{
	ios::sync_with_stdio(0);
	cin.tie(0);
	cin >> n;
	for (int i = 0; i < n; i++)
		cin >> arr[i];
	sort(arr, arr + n, cmp);
	for (int i = 0; i < n; i++)
	{
		if (i != 0 && arr[i - 1] == arr[i])
			continue;
		cout << arr[i] << '\n';
	}

	return 0;
}