알고리즘 공부 및 문제 풀이/백준(BOJ)
[c++] 백준 1181 단어 정렬
yoonjiy
2021. 7. 18. 16:31
백준 단계별로 풀어보기 [정렬] 단어 정렬
https://www.acmicpc.net/problem/1181
1181번: 단어 정렬
첫째 줄에 단어의 개수 N이 주어진다. (1 ≤ N ≤ 20,000) 둘째 줄부터 N개의 줄에 걸쳐 알파벳 소문자로 이루어진 단어가 한 줄에 하나씩 주어진다. 주어지는 문자열의 길이는 50을 넘지 않는다.
www.acmicpc.net
[풀이]
string을 저장하는 배열을 만들고, 길이가 짧은 순으로, 만약 길이가 같다면 사전 순으로 정렬한다. cmp 함수에 이와 같은 조건을 정해주고 sort 함수 세번째 인자로 전달해준다. 정렬 후 같은 단어가 있다면 한 번만 출력하도록 continue를 걸어준다.
[코드]
#include <iostream>
#include <algorithm>
bool cmp(const std::string& a, const std::string& b) {
//길이가 짧은 것부터, 같으면 사전 순으로.
if (a.length() == b.length()) {
return a < b;
}
else return a.length() < b.length();
}
int main() {
int n;
std::cin >> n;
std::string s;
std::string* arr = new std::string[n];
for (int i = 0; i < n; i++) {
std::cin >> arr[i];
}
sort(arr, arr + n, cmp);
for (int i = 0; i < n; i++) {
if (arr[i] == arr[i + 1])
continue;
std::cout << arr[i] << "\n";
}
return 0;
}