[C++][BOJ/백준] 1181 단어 정렬

 

 

Baekjoon Online Judge

1181번: 단어 정렬

 

 

[문제]

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

     1. 길이가 짧은 것부터

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

단, 중복된 단어는 하나만 남기고 제거해야 한다.

 

 

[입력]

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

 

 

[출력]

조건에 따라 정렬하여 단어들을 출력한다.

 

 

---

 

[풀이]

단어를 문자열 배열에 입력받고 sort() 함수와 compare() 함수를 사용해서 정렬하면 된다.

compare() 함수는 인자로 받은 string 변수 a와 b의 길이를 비교하는데

만약 a와 b의 길이가 같다면 'return a < b'를 통해 사전 순으로 정렬되도록 하고,

a와 b의 길이가 다르다면 'return a.length() < b.length()'를 통해 길이가 짧은 것부터 정렬되도록 한다.

 

정렬이 끝나고 단어들을 출력할 때는 중복된 단어가 하나만 출력되도록

if문을 통해 res[i]와 res[i - 1]이 다를 때만 단어를 출력하게 한다.

 

 

[코드]

#include <iostream>
#include <algorithm>
#include <string>
using namespace std;

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

int main()
{
    int N;
    string res[20000];

    cin >> N;

    for (int i = 0; i < N; i++)
        cin >> res[i];

    sort(res, res + N, compare);

    for (int i = 0; i < N; i++)
    {
        if (res[i] != res[i - 1])
            cout << res[i] << '\n';
    }

    return 0;
}