[백준] 1181번: 단어 정렬

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

1181번: 단어 정렬

문제 이해

이것도 그냥 compare함수를 잘 설계하면 되는 문제이다.

 

작성 코드

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

using namespace std;

int N;
vector<string> v;

bool compare(string a, string b){

    if(a.length() < b.length()) return true;
    else if(a.length() == b.length()){
        if(a<b) return true;
        else return false;
    }
    else return false;
}

int main(void){
    cin >> N;
    for(int i = 0; i<N; i++){
        string tmp;
        cin >> tmp;
        v.push_back(tmp);
    }

    sort(v.begin(), v.end(), compare);

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

    return 0;
}