[알고리즘 연습] 메뉴 리뉴얼 (프로그래머스 lv2, 스위프트)

문제

https://programmers.co.kr/learn/courses/30/lessons/72411

코딩테스트 연습 - 메뉴 리뉴얼

풀이

import Foundation

func solution(_ orders:[String], _ course:[Int]) -> [String] {
​​​​var result = [String]()

​​​​for n in course {
​​​​​​​​var temp = [String:Int]()
​​​​​​​​var check = true

​​​​​​​​orders.forEach { order in
​​​​​​​​​​​​combination(order, n).forEach { 
​​​​​​​​​​​​​​​​temp[$0, default: 0] += 1
​​​​​​​​​​​​​​​​if temp[$0]! >= 2 {
​​​​​​​​​​​​​​​​​​​​check = false
​​​​​​​​​​​​​​​​}
​​​​​​​​​​​​}
​​​​​​​​}

​​​​​​​​if check { continue }

​​​​​​​​let sortedTemp = temp.sorted(by: { $0.value > $1.value })
​​​​​​​​let limit = sortedTemp[0].value

​​​​​​​​for (key, value) in sortedTemp {
​​​​​​​​​​​​if value == limit{
​​​​​​​​​​​​​​​​result.append(key)
​​​​​​​​​​​​} else {
​​​​​​​​​​​​​​​​break
​​​​​​​​​​​​}
​​​​​​​​}
​​​​}

​​​​return result.sorted()
}

func combination(_ str: String, _ n: Int) -> [String] {
​​​​var result = [String]()

​​​​if str.count < n {
​​​​​​​​return result
​​​​}

​​​​var sa = Array(str).map { String($0) }
​​​​var stack = [(String, Int)]()

​​​​for i in 0..<str.count {
​​​​​​​​stack.append((sa[i],i))
​​​​}

​​​​while stack.count > 0 {
​​​​​​​​let now = stack.removeLast()

​​​​​​​​let nowString = now.0
​​​​​​​​let nowIdx = now.1

​​​​​​​​if nowString.count == n {
​​​​​​​​​​​​result.append(String(nowString.sorted()))
​​​​​​​​​​​​continue
​​​​​​​​}

​​​​​​​​guard nowIdx+1 <= sa.count-1 else { continue }

​​​​​​​​for i in nowIdx+1...sa.count-1 {
​​​​​​​​​​​​stack.append((nowString + sa[i], i))
​​​​​​​​}
​​​​}

​​​​return result
}

• 어렵지 않은 문제였지만, Swift에서 조합을 사용하는 것이 익숙치않아 포스팅해놓습니다.
• 가능한 조합의 케이스를 모두 확인하여 return하는 함수를 짜놓고 이를 활용하는 방식으로 해결했습니다.
• combination을 Generic하게 구현해놓고 필요할때마다 사용하는 것도 좋은 전략이 될 것 같습니다.