백준 1181 단어 정렬

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

1181번: 단어 정렬

 

 

 

EXPLANATION

 

find 함수를 이용해 중복되는 단어를 찾아 제외시키면서 배열을 초기화하고

sort함수를 이용해 재배열하면된다.

 

 

 

CODE

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

using namespace std;

bool compareWith(string a, string b) {
    if(a.length() == b.length())   //단어 길이가 같으면 사전 순으로
        return a < b;
    return a.length() < b.length(); //단어 길이가 다르면 길이가 짧은 것부터 
}

int main(){
    int N;
    cin >> N;
    vector<string> arr;
    for(int i = 0; i < N; i++){
        string str;
        cin >> str;
        if(find(arr.begin(), arr.end(),str) == arr.end()){ //같은 단어 중복 입력 제거, find 함수는 해당 단어가 존재하지 않으면 vector.end()를 반환
            arr.push_back(str);
        }
    }
    sort(arr.begin(), arr.end(),compareWith);

    for(int i=0; i<arr.size(); i++){
        cout<<arr[i]<<"\n";
    }
    return 0;
}