백준 1181 : 단어 정렬

문제 링크입니다 https://www.acmicpc.net/problem/1181

1181번: 단어 정렬

알파벳 소문자로 이루어진 N개의 단어가 입력으로 주어지면 단어들을 조건에 따라 정렬하여 출력하는 문제였습니다. 단, 같은 단어가 여러 번 입력된 경우에는 한 번씩만 출력해야 했습니다. 조건은 다음과 같고,

 

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

단어를 입력받고 string 벡터에 모든 단어를 push하고, 위 조건에 맞게 compare 함수를 만들어 정렬한 후 처음부터 끝까지 직전에 출력하지 않았으면 출력하는 형태로 문제를 해결했습니다.

 

 

[소스코드]

 

#include<algorithm>
#include<iostream>
#include<string>
#include<vector>
using namespace std;
 
int n;
 
// 정렬을 위한 compare 함수
bool comp(string a, string b) {
 
    // 길이가 같으면 사전순으로 앞서는 것을 true로 반환
    if (a.length() == b.length()) return a < b;
 
    // 길이가 다르면 길이가 짧은 것을 true로 반환
    else return a.length() < b.length();
}
 
int main(void) {
    cin >> n;
 
    // 각 단어들을 저장할 벡터
    vector <string> s;
 
    while (n--) {
        string ss;
        cin >> ss;
 
        // 단어를 입력받고 s 벡터에 차례차례 넣습니다
        s.push_back(ss);
    }
    // 조건에 맞게 만든 comp 함수를 통해 정렬합니다
    sort(s.begin(), s.end(), comp);
    
    // 직전에 출력한 단어를 저장할 변수
    string prev;
 
    // s 벡터의 처음부터 끝까지
    for (string a : s) {
        // 전에 출력한 단어가 아니라면 출력합니다
        if(prev != a) cout << a << '\n';
        prev = a;
    }
    return 0;
}