[Algorithm] 모의고사 (완전탐색)

🖥 문제

수포자는 수학을 포기한 사람의 준말입니다. 수포자 삼인방은 모의고사에 수학 문제를 전부 찍으려 합니다. 수포자는 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 함수를 작성해주세요.

🤔 풀이 방법

각 수포자가 찍는 방식을 담은 3개의 배열과 3명의 점수를 담을 score 배열을 하나 만들어준다.
배열의 0번째는 1번 수포자의 점수, 1번째는 2번 수포자의 점수, 2번째는 3번 수포자의 배열을 저장할 것이다.

for문을 돌며 answers와 3명의 답을 각각 비교하여 같을 경우 score에 1을 더해준다.
이때 % 연산자를 사용하여 0, 1, 2, 3, 4, 5, 0, 1, 2, 3, 4, ... 와 같은 순서로 비교한다.
답을 다 비교한 후에 가장 많은 문제를 맞힌 개수를 max에 저장한다.
그 후 다시 for문을 돌며 max개의 답을 맞힌 수포자의 번호를 저장하여 이를 리턴하였다.

딕셔너리를 사용하여 풀면 더 직관적이게 코드를 짤 수 있을 것 같다.

👩🏻‍💻 Swift Code

import Foundation

func solution(_ answers:[Int]) -> [Int] {
    var person1 = [1, 2, 3, 4, 5]
    var person2 = [2, 1, 2, 3, 2, 4, 2, 5]
    var person3 = [3, 3, 1, 1, 2, 2, 4, 4, 5, 5]
    var score = [0, 0, 0]
    var answer: [Int] = []

    for i in 0..<answers.count {
        if answers[i] == person1[i % 5] {
            score[0] += 1
        }
        if answers[i] == person2[i % 8] {
            score[1] += 1
        }
        if answers[i] == person3[i % 10] {
            score[2] += 1
        }
    }

    let max = score.max()!

    for i in 0..<score.count {
        if score[i] == max {
            answer.append(i + 1)
        }
    }

    return answer
}

 

코딩테스트 연습 - 모의고사