완전탐색 - 프로그래머스(Swift)

완전탐색 - 프로그래머스(Swift)

 

 

 문제 설명

 

https://school.programmers.co.kr/learn/courses/30/lessons/42840

프로그래머스

 

 

 나의 풀이

 

cnt 배열을 튜플로 구성해서 인덱스에 접근해 증가시키는 방법을 사용했다. 

if문을 많이 사용하고 좀 복잡하게 인덱스에 접근한 것 같아서 풀고 나서도 굉장히 찝찝했다..

 

import Foundation

func solution(_ answers:[Int]) -> [Int] {
    let arrs = [
        (1, [1, 2, 3, 4, 5]),
        (2, [2, 1, 2, 3, 2, 4, 2, 5]),
        (3, [3, 3, 1, 1, 2, 2, 4, 4, 5, 5])
    ]
    
    var idx = 0
    var cnts = [(Int, Int)](repeating: (0, 0), count: 4)
    
    for i in 1..<4 {
        cnts[i].0 = i
    }
    for arr in arrs {
        answers.forEach { (num) in
            if num == arr.1[idx] {
                cnts[arr.0].1 += 1
            }
            idx += 1
            if idx == arr.1.count {
                idx = 0
            }
        }
        idx = 0
    }
    cnts.remove(at: 0)
    var max = 0
    cnts.forEach { (cnt) in
        if cnt.1 > max {
            max = cnt.1    
        }   
    }
    
    var fillArr = cnts.filter{ $0.1 == max }
    
    if fillArr.count == 1{
        return [fillArr[0].0]
    } else {
        return fillArr.sorted(by: {$0.0 < $1.0}).map{$0.0}
    }
}

 

 

 다른 사람의 풀이

 

딕셔너리를 이용한 깔끔한 풀이이다.

배열안에 튜플을 넣었던 나와 달리 이름있는 튜플에 배열을 넣는 구조이다.

그리고 배열의 개수만큼 for문을 도는 것이 아닌 문제를 탐색할때마다 각각에 if문을 주어 한 번에 처리하는 게 신기했다.

cnt배열을 사용할 때 딕셔너리를 이용하면 굉장히 유용할 것 같다고 느꼈다.

 

import Foundation

func solution(_ answers:[Int]) -> [Int] {
    let answer = (
        a: [1, 2, 3, 4, 5],
        b: [2, 1, 2, 3, 2, 4, 2, 5],
        c: [3, 3, 1, 1, 2, 2, 4, 4, 5, 5]
    )
    
    var points = [1:0, 2:0, 3:0]
    
    for (idx, val) in answers.enumerated() {
        if val == answer.a[idx % 5] { points[1]! += 1}
        if val == answer.b[idx % 8] { points[2]! += 1}
        if val == answer.c[idx % 10] { points[3]! += 1}
    }
    
    return points.sorted(by: {$0.key < $1.key}).filter{ points.values.max() == $0.value }.map{$0.key}
}

 

 

 피드백

 

1. 뭔가 문제가 복잡하다고 느낄 때는 새로운 자료구조를 사용하려고 노력해보자.

2. 인덱스가 초과되고 원점으로 돌아올 때는 나머지 연산자를 사용해서 푸는 경우가 많다. 

3. 딕셔너리.values.max()를 하게 되면 value에 해당하는 max값이 추출된다. 

4. 딕셔너리에서 값[key]로 접근하면 옵셔널이 나오기 때문에 언래핑한 상태로 구현해야한다.