[C++ / 1181] 단어 정렬

문제

1181번: 단어 정렬

---

문제 설명

아래와 같은 조건으로 나열하면 된다.

1. 길이가 짧은 것 부터

2. 길이가 같다면 사전 순으로

3. 중복은 제거한다.

---

풀이 방식

알고리즘 적으로 적어둘만 한건 없는 것 같다.

---

솔루션

#include <iostream>
#include <algorithm>
#include <vector>

using namespace std;

bool compare(string a, string b)
{
  if (a.length() != b.length())
    return a.length() < b.length();
  else
    return a < b;
}

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

  int n;
  cin >> n;

  vector<string> input(n);

  for (int i = 0; i < n; i++)
    cin >> input[i];

  sort(input.begin(), input.end(), compare);

  for (int i = 0; i < n; i++)
    if (input[i] == input[i + 1])
      continue;
    else
      cout << input[i] << endl;
}

---

알게된 점

C++의 Algorithm 라이브러리에 들어있는 sort() 함수는 다음과 같이 사용할 수 있다.

sort(v.begin(), v.end())

위와 같이 사용한다면, 길이는 고려하지 않고 사전순으로 정렬된 결과를 얻을 수 있다.

이 sort() 함수의 argument를 다음과 같이 3개로 받으면,

 sort(v.begin(), v.end(), comp)

comp 부분의 결과 값에 따라 정렬 결과가 달라지는데,

comp를 구현할 때, a < b의 경우 오름차순, a > b의 경우 내림차순으로 구현된다.