백준 문제풀이

[beakjoon-1181-C++] 단어 정렬

박여치 2023. 1. 19. 16:06

문제

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

  1. 길이가 짧은 것부터
  2. 길이가 같으면 사전 순으로

입력

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

출력

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

코드

sort함수의 3번째 인자로 정렬 기준을 넣을 수 있음을 활용하였다.

compare 함수를 보면 길이가 같을 경우, 작은 순서대로 길이가 다를 경우, 짧은 순서대로 정렬 기준을 잡은 것을 볼 수 있다. 

find 함수를 통해 입력 받을 시, 같은 단어는 넣지 않고 넘어감으로써 같은 단어는 1번만 출력되게 하였습니다.