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

블로그에 올리는 모든 문제 풀이는 깃허브에 올려져 있습니다.

---

문제 설명

• 알파벳 소문자로 이루어진 N개의 단어가 들어오면 아래와 같은 조건에 따라 정렬하는 프로그램을 작성하시오.
  1. 길이가 짧은 것부터
  2. 길이가 같으면 사전 순으로

---

풀이 전략

• algorithm의 sort와 vector를 이용해서 정렬한다.
• 정렬 기준은 사이즈가 같으면 두 값을 비교, 다르면 오름차순 정렬한다.

소스 코드

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

using namespace std;
bool Compare( string a, string b )
{
    if (a.size() == b.size())
    {
        int x = a.compare(b);
        return x < 0;
    }
    else
        return (a.size() < b.size());
}

int main()
{
    int x;
    vector<string> vec;
    cin >> x;

    for (int i = 0; i < x; i++)
    {
        string s;
        cin >> s;
        vec.push_back(s);
    }

    sort(vec.begin(), vec.end(), Compare);
    vec.resize(unique(vec.begin(), vec.end()) - vec.begin());

    for (int i = 0; i < vec.size(); i++)
    {
        cout << vec[i] << '\n';
    }
}

---

링크

https://www.acmicpc.net/problem/1181

1181번: 단어 정렬