[선택] 1181 단어정렬

정답 코드

let count = Int(readLine()!)!
var set: Set<String> = Set<String>()

for _ in 0..<count {
    set.insert(readLine()!)
}

let sortedSet = set.sorted { (first, second) -> Bool in
    if first.count == second.count {
        return first < second
    } else {
        return first.count < second.count
    }
}

for component in sortedSet {
    print(component)
}

 

코드 설명

• 중복이 된 내용은 한번만 출력해주기 때문에 set을 생성합니다.
• sorted(by:) 메서드를 사용하여 길이가 작은 순 대로 정렬하고 길이가 같다면 알파벳 순으로 정렬합니다.

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

1181번: 단어 정렬