[BOJ] 백준 1181번 단어 정렬 풀이

| |
백준 온라인 저지 백준 1181번 - 단어 정렬 C++ 풀이


문제

알파벳 소문자로 이루어진 N개의 단어가 들어오면 아래와 같은 조건에 따라 정렬하는 프로그램을 작성하시오.

  • 길이가 짧은 것부터
  • 길이가 같으면 사전 순으로


입력

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


출력

조건에 따라 정렬하여 단어들을 출력한다. 단, 같은 단어가 여러 번 입력된 경우에는 한 번씩만 출력한다.


풀이

문제에서 특정한 문제해결능력이나 알고리즘을 요구하기보단 자신이 사용하는 언어의 기능과 문법을 완벽히 숙지하고 있는지를 테스트하는 문제에 가깝다. 문제에서 요구하는 구현은 다음과 같이 두가지이다.

  • 정렬 조건을 직접 설정하여 정렬하기
  • 중복된 요소 처리하기

두가지 기능 모두 거의 모든 언어에서 기본적으로 제공하는 기능이다. C++ 의 경우 STL을 통해 충분히 구현할 수 있다. 정렬 조건은 sort 함수의 세번째 인자로 직접 구현한 정렬 조건을 넣어주면 되고, 중복 요소의 경우 vector의 정렬 이후 erase 함수와 unique 함수를 통해 한 번에 처리할 수 있다.


전체 코드

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
#include <iostream>
#include <string>
#include <vector>
#include <algorithm>
using namespace std;

// 정렬 조건
bool pred(string& a, string& b){
   if(a.size()==b.size()) return a<b;
   return a.size()<b.size();
}

int main(){
   ios_base::sync_with_stdio(0);
   cin.tie(0); cout.tie(0);

   int N; cin>>N;
   vector<string> v(N);

   while(N--) cin>>v[N];

   sort(v.begin(),v.end(),pred); // 정렬
   v.erase(unique(v.begin(),v.end()),v.end()); // 중복 처리

   for(string s:v) cout<<s<<'\n';

   return 0;
}
0%