[Swift 알고리즘] Programmers 모의고사

Level 1 완전탐색

모의고사

문제 설명
수포자는 수학을 포기한 사람의 준말입니다. 수포자 삼인방은 모의고사에 수학 문제를 전부 찍으려 합니다. 수포자는 1번 문제부터 마지막 문제까지 다음과 같이 찍습니다.
 1번 수포자가 찍는 방식: 1, 2, 3, 4, 5, 1, 2, 3, 4, 5, ...
 2번 수포자가 찍는 방식: 2, 1, 2, 3, 2, 4, 2, 5, 2, 1, 2, 3, 2, 4, 2, 5, ...
 3번 수포자가 찍는 방식: 3, 3, 1, 1, 2, 2, 4, 4, 5, 5, 3, 3, 1, 1, 2, 2, 4, 4, 5, 5, ...
1번 문제부터 마지막 문제까지의 정답이 순서대로 들은 배열 answers가 주어졌을 때, 가장 많은 문제를 맞힌 사람이 누구인지 배열에 담아 return 하도록 solution 함수를 작성해주세요.
제한 조건
- 시험은 최대 10,000 문제로 구성되어있습니다.
- 문제의 정답은 1, 2, 3, 4, 5중 하나입니다.
- 가장 높은 점수를 받은 사람이 여럿일 경우, return하는 값을 오름차순 정렬해주세요.

 

입출력 예시

print(solution([1,2,3,4,5])) //[1]
print(solution([1,3,2,4,2])) //[1,2,3]

 

내 답안

import Foundation

func solution(_ answers:[Int]) -> [Int] {
    let arr = Array(0...answers.count-1)
    let first = arr.map { [1,2,3,4,5][$0 % 5] }
    let second = arr.map { ($0 % 2 == 0) ? 2 : [1,3,4,5][$0/2 % 4] }
    let third = arr.map { [3,3,1,1,2,2,4,4,5,5][$0 % 10] }
    
    let score = [arr.filter{ answers[$0] == first[$0] }.count, arr.filter{ answers[$0] == second[$0] }.count, arr.filter{ answers[$0] == third[$0] }.count]
    
    var result = [Int]()
    for i in 0...2 {
        if score[i] == score.max() {
            result.append(i+1)
        }
    }
    return result
}

처음에는 score를 [1:점수, 2:점수, 3:점수] 이렇게 dictionary를 사용해서 풀어보려 했는데 그렇게 하는게 더 복잡해지는 것 같아서 그냥 배열로 for문 돌려서 순서대로 append 해주는 방식으로 해결을 했다. 

에고.. 너무 복잡한가..? 했는데 그래도 한번에 통과해서 뿌듯- 😊

 

내 풀이 

1. 입력받은 문제 정답의 길이만큼 1번, 2번, 3번 수포자의 답안을 생성한다.

2. 각 답안에서 정답의 갯수를 담아 score 배열을 생성한다.

3. 최고 점수를 받은 수포자들을 result 배열에 append 해준다. 

4. result를 반환한다. 

 

 

다른 답안

import Foundation

func solution(_ answers:[Int]) -> [Int] {
    let answer = (
        a: [1, 2, 3, 4, 5], // index % 5
        b: [2, 1, 2, 3, 2, 4, 2, 5], // index % 8
        c: [3, 3, 1, 1, 2, 2, 4, 4, 5, 5] // index % 10
    )
    var point = [1:0, 2:0, 3:0]

    for (i, v) in answers.enumerated() {
        if v == answer.a[i % 5] { point[1] = point[1]! + 1 }
        if v == answer.b[i % 8] { point[2] = point[2]! + 1 }
        if v == answer.c[i % 10] { point[3] = point[3]! + 1 }
    }

    return point.sorted{ $0.key < $1.key }.filter{ $0.value == point.values.max() }.map{ $0.key }
}

다른 분이 딕셔너리 사용해서 푼 답안이 있더라. 

enumerated() 는 아직 사용해본 적이 없어서 뭔지 알아봐야겠다.