백준 1181번: 단어 정렬 (Python, C++)

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

1181번: 단어 정렬

---

Level: Silver V

Solved By: Sort

 

파이썬에서 다중 조건에 대해 정렬하는 방법은 A라는 상위 조건과 B라는 하위 조건이 있으면, B로 먼저 정렬을 한 후에 A로 정렬을 합니다. (Ref. https://velog.io/@1204jh/1181)

 

또한, 중복 값들을 없애주어야 하기 때문에 입력받은 리스트를 set으로 한 번 전환하여 중복 값들을 버린 후에 리스트로 다시 담아 정렬을 합니다.

 

상위 조건과 하위 조건에 대해 궁금해서 C++로도 파이썬 코드 밑에 풀어보았습니다.

-> 파이썬과 똑같음, 애초에 논리적으로 접근했을 때, 저 말이 맞음

n = int(input())
li = []

for i in range(0, n):
    a = input()
    li.append(a)

set_li = set(li)
li = list(set_li)

li.sort()
li.sort(key=len)

for a in li:
    print(a)

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;

vector<string> v;

bool cmp(string a, string b){
    if(a.length() == b.length()) return a < b; // 하위 조건
    else return a.length() < b.length(); // 상위 조건
}

int main(){
    int N; cin >> N;
    for(int i = 0; i < N; i++){
        string a; cin >> a;
        v.push_back(a);
    }
    
    sort(v.begin(), v.end(), cmp);
    v.erase(unique(v.begin(), v.end()), v.end());
    
    for(auto a : v){
        cout << a << '\n';
    }
    
    return 0;
}