[BAEKJOON] 1181 - 단어 정렬 (C++)

문제 링크
난이도 : Silver 5
(solved.ac 2023.09.27. 기준)

문제

알파벳 소문자로 이루어진 N개의 단어가 들어오면 아래와 같은 조건에 따라 정렬하는 프로그램을 작성하시오.
1. 길이가 짧은 것부터
2. 길이가 같으면 사전 순으로.
단, 중복된 단어는 하나만 남기고 제거해야 한다.

알고리즘

정렬

풀이

입력받은 문자열을 정렬하기 위해 C++ 표준라이브러리인 <algorithm>에 있는 sort 함수를 사용한다. 하지만 단순히 사전순으로 정렬하는 것이 아닌, 조건에 따라 정렬해야 하기 때문에 compare을 sort의 사용자 지정 함수로 만들어 정렬한다. 그리고 중복되는 단어는 하나만 남기고 제거해야 하기 때문에 첫번째 단어는 출력하고, 두번째 단어부터는 중복되는 단어들을 출력하지 않는다.

코드

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
bool compare(string a, string b) // sort 정렬 함수
{
    if (a.size() == b.size()) // a와 b의 길이가 같을 때
    {
        return a < b; // 사전순
    }
    return a.size() < b.size(); // 길이가 작은 순
}
int main()
{
    int n;
    cin >> n;
    vector<string> list(n);
    for (int i = 0; i < n; i++)
    {
        cin >> list[i];
    }
    sort(list.begin(), list.end(), compare);  //compare에 따라 정렬
    for (int i = 0; i < n; i++)
    {
        if (i == 0)  //첫번째 문자열은 출력
        {
            cout << list[i] << "\n";
            continue;
        }
        if (list[i] != list[i - 1]) // 중복되는 문자열 출력 X
        {
            cout << list[i] << "\n";
        }
    }
    return 0;
}