[Baekjoon] 1181 : 단어 정렬 (C++)

Baekjoon Online Judge 1181번 : 단어 정렬

---

문제

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

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

입력

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

출력

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

예제 입력

13
but
i
wont
hesitate
no
more
no
more
it
cannot
wait
im
yours

예제 출력

i
im
it
no
but
more
wait
wont
yours
cannot
hesitate

---

코드

다른 문제를 풀면서 익힌 sort()함수의 compare 인수를 활용하여 문제에서 주어진 조건을 만족하도록 정렬하였다. 다음에 기회가 된다면 입력하고 정렬하는 것이 아니라 입력 과정에서 정렬을 하는 방식으로도 시도해봐야겠다.

#include <iostream>
#include <string> // string 자료형
#include <algorithm> // sort()

using namespace std;

bool cmp_str(string s1, string s2) // 정렬 기준
{
  if(s1.length() == s2.length()) // 길이가 같을 때, 사전순
    return s1 < s2;
  else // 길이가 짧은 순
    return s1.length() < s2.length();
}

int main(void)
{
  int N;
  string* words;

  cin >> N;
  words = new string[N]; // 크기가 N인 배열 동적할당

  for(int i = 0; i < N; i++)
    cin >> words[i]; // 주어진 단어들을 배열에 입력

  sort(words, words+N, cmp_str); // 주어진 조건에 맞게 정렬

  cout << words[0] << '\n';
  for(int i = 1; i < N; i++)
    if(words[i].compare(words[i-1]) != 0) // 중복되는 단어 출력 제외
      cout << words[i] << '\n';

  delete[] words; // 동적할당 해제
  return 0;
}