[C++] 백준 1181번 단어정렬 문제풀이

문제 링크

백준 1181번 단어정렬

1181번: 단어 정렬

 

간단한 심화 정렬 문제이다. 

string으로 단어를 받고 sort 함수를 이용해 정렬을 하는데 compare 함수를 따로 길이순으로 반환을 한 후에 

길이가 같다면 알바벳순으로 오름자순 정렬을 하는 형식이다. 

string size 함수를 이용해 길이를 파악하고 길이가 같다면 단순 비교를 통해 함수를 만들었다.

정렬을 공부하는 사람이라면 도움이 많이 될 것이다.

 

#include <iostream>
#include <vector>
#include <algorithm>
 
using namespace std;
 
bool compare(string a,string b){
    if(a.size()==b.size()){
        return a<b;
    }else{
        return a.size()<b.size();
    }
}
int main(){
    
    vector <string> a;
    string temp;
    int N;
    cin>>N;
    
    for(int i=0;i<N;i++){
        string word;
        cin>>word;
        a.push_back(word);        
    }
    
    sort(a.begin(),a.end(),compare);
    
    for(int i=0;i<N;i++){
        if(temp==a[i])continue;
        cout<<a[i]<<'\n';
        temp=a[i];
    }
    return 0;
}