[백준 알고리즘/BOJ/C++] 1181 단어 정렬

 

 

알파벳 소문자로 이루어진 단어들을 입력받으면 길이 순으로, 길이가 같으면 사전 순으로 정렬하는 문제

 

이거야 말로 벡터를 쓰면 좋은 문제다 (11650, 11651 참고)

벡터는 기본 first 기준으로, 같다면 second 기준으로 정렬하기 때문

그러므로 first엔 단어의 길이를, second엔 단어를 넣어주고 정렬하면 된다

 

 

#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
 
int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);
 
    int n;
    cin >> n;
 
    vector<pair<int, string>> arr(n); //첫번째는 길이 두번째는 문자
 
    for (int i = 0; i < n; i++) {
        cin >> arr[i].second;
        arr[i].first = arr[i].second.length();
    }
 
    sort(arr.begin(), arr.end()); //길이 같으면 자동으로 second 기준!!
 
    cout << arr[0].second << "\n";
    for (int j = 1; j < n; j++) {
        if(arr[j].second != arr[j-1].second)
            cout << arr[j].second << "\n";
    }
}

 

벡터만 쓸 줄 안다면 쉬운 문제