알고리즘
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;
}