[swift] 프로그래머스 - 숫자 짝꿍

프로그래머스

문제 푸는 데 있어 도움이 되도록 나의 풀이와 개선된 풀이를 올립니다.
또한 풀이 후 다른 사람의 풀이를 보고 참고할만한 풀이도 올립니다.
- 문제에 따라 나의 풀이만 있을 수 있습니다.
- 해당 풀이들은 풀이 중 하나일 뿐 최선의 풀이는 아닐 수 있습니다.

 

---

 

문제 설명

 

 1. 두 문자열을 비교하여 모두 포함하는 문자를 찾는다.

2. 포함하는 횟수만큼 리턴한다(문자열로).

 3. 찾은 문자가 "0"으로만 구성되어 있다면 "0"을 반환한다.

 4. 모두 포함하는 문자가 없다면 "-1"을 반환한다.

 5. 찾은 문자는 내림차순으로 정렬한다.

 

 

---

 

첫 번째 풀이(실패)

import Foundation

func solution(_ X:String, _ Y:String) -> String {
    var myX = X.filter { Y.contains($0) }.sorted(by: >)
    var myY = Y.filter { X.contains($0) }.sorted(by: >)
    var result = [Character]()

    if myY.isEmpty && myX.isEmpty { return "-1" }

    for i in myX {
        if myY.contains(i) {
            result.append(i)
            myY.remove(at: myY.firstIndex(of: i)!)
        }
    }
    if result.reduce(0, { $0 + Int(String($1))! }) == 0 { return "0" }
    return String(result)
}

 

첫 번째 풀이입니다.

시간제한에 걸려 실패했습니다.

 

---

 

두 번째 풀이(실패)

import Foundation

func solution(_ X:String, _ Y:String) -> String {
    
    var myX = [Character: Int]()
    var result = [String]()
    
    for i in X {
        if myX[i] == nil { myX[i] = 1 }
        else { myX[i]! += 1 }
    }
    
    for i in Y {
        if myX[i] == nil || myX[i] == 0 { continue }
        else { result.append(String(i)); myX[i]! -= 1 }
    }
    
    if result.isEmpty { return "-1" }
    if result.allSatisfy({ $0 == "0" }) { return "0" }
    return result.sorted(by: >).joined()
}

 

고차함수를 빼고

최대한 간결하게 코드를 짜보려 노력했지만

역시 시간제한에 걸렸습니다.

 

---

 

세 번째 풀이(정답)

import Foundation

func solution(_ X:String, _ Y:String) -> String {
    
    var myX = [Character: Int]()
    var myY = [Character: Int]()
    var result = [String]()
    
    for i in X { myX[i] = (myX[i] ?? 0) + 1 }
    for i in Y { myY[i] = (myY[i] ?? 0) + 1 }
    
    for i in myX.keys {
        result += Array(repeating: String(i), count: min(myX[i] ?? 0, myY[i] ?? 0))
    }
    
    if result.isEmpty { return "-1" }
    if result.allSatisfy({ $0 == "0" }) { return "0" }
    
    return result.sorted(by: >).joined()
}

 

Y도 딕셔너리로 만들어봤습니다.

 

두번째 풀이에서는

Y를 딕셔너리로 만들어서 푸는거보다

Y를 탐색하며 한번에 X를 비교해 푸는게 빠르다 생각했는데

 

오히려

Y 딕셔너리를 하나 만들고

X 딕셔너리를 탐색하며 비교했더니

시간제한에 통과할 수 있었습니다.

 

---

 

참고할만한 풀이

import Foundation

func solution(_ X:String, _ Y:String) -> String {
    let arr1 = Array(X)
    let arr2 = Array(Y)
    let uniqueKey = Set(arr1).intersection(Set(arr2))
    var answer = ""
    for key in uniqueKey.sorted(by: >) {
        answer += String(repeating: key, count: min(arr1.filter{$0 == key}.count, arr2.filter{$0 == key}.count))
    }
    return answer == "" ? "-1" : ((uniqueKey.count == 1 && uniqueKey.contains("0")) ? "0" : answer)
}

 

좋아요가 많았던 풀이인데요.

intersection함수를 처음 봐서 가져와봤습니다.

 

let uniqueKey = Set(arr1).intersection(Set(arr2))

 

해당함수는

Set(arr1)와 Set(arr2)를 비교해

중복되는 값을 새로운 Set으로 리턴해줍니다.

 

하지만 시간제한에 예민한 11~15번 테스트케이스에서

약 9초가 나오면서 아슬아슬하게 통과하는 모습을 보입니다.