[BOJ/C++]1181번_단어 정렬

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

 

##문제 풀기 전 내가 알고 있었어야 할 개념: (작성중)

 

 

#include <iostream>
#include <string>
#include <vector>
#include <algorithm>
#define SIZE 20000
 
using namespace std;
 
void init() {
 
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
}
 
bool compare(string a, string b) {
    if (a.size() == b.size()) {
        return a < b;
    }
    else {
        return a.size() < b.size();
    }
}
 
int main() {
 
    init();
 
    int N;
    string input;
    vector<string> V;
    vector<string>::iterator it;
    cin >> N;
 
    for (int i = 0; i < N; i++) {
        cin >> input;
        V.push_back(input);
    }
 
    sort(V.begin(), V.end(), compare);
    /* 나의 틀린 코드 :
    for (int i = 0; i < N; i++) {
        cout << V[i] << '\n';
        if (i != N - 1) {
            if (V[i] == V[i + 1]) {
                it = V.begin() + i + 1;
                it = V.erase(it);
                N--;
            }
        }
    }
    */
 
    //수정 후
    // 중복 원소 제거
    auto last = unique(V.begin(), V.end());
    V.erase(last, V.end());
    // 결과 출력
    for (const string& s : V) {
        cout << s << '\n';
    }
 
    return 0;
}