[Swift] 프로그래머스(lv.1) 12 - 모의고사

모의고사

문제 설명

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

 

 

문제 풀이

1. 우선 수포자가 찍은 답을 배열로 선언했다.
2. 정답이랑 수포자들의 답을 비교하는데, 수포자들의 답이 반복되므로 배열의 크기로 나눈 나머지와 비교해준다.
3. 답을 비교해 같을 경우 결과 배열의 값을 증가시켜준다.
4. 배열에서 max()메소드를 사용해 가장 큰 값을 찾는다.
5. 가장 큰 값과 같은 값을 가지는 index를 찾아 index+1 = 학생번호 을 result 배열에 추가해준다. 

2번 방법을 알아야 이 문제를 풀 수 있다.

 

코드

내 코드

import Foundation

func solution(_ answers:[Int]) -> [Int] {
    
    let studentOne = [1,2,3,4,5]
    let studentTwo = [2,1,2,3,2,4,2,5]
    let studentThree = [3,3,1,1,2,2,4,4,5,5]    
    let problemCount = answers.count
    var resultCount = [0, 0, 0]
    var result:[Int] = []
    
    for i in 0..<problemCount {
        if (answers[i] == studentOne[i % studentOne.count]) {
            resultCount[0] += 1
        }
        if (answers[i] == studentTwo[i % studentTwo.count]) {
            resultCount[1] += 1
        }
        if (answers[i] == studentThree[i % studentThree.count]) {
            resultCount[2] += 1
        }
    }
    
    let max = resultCount.max()
    for i in 0..<3 {
        if(max == resultCount[i]){
            result.append(i+1)
        }
    }
    
    return 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 }
}

가장 깔끔한 코드는 딕셔너리를 사용했다.

훨씬 코드가 깔끔해졌다.

 

 

 

후기

반복되는 값은 항상 나머지를 이용하자

Dictonary 를 사용할 수 있는 상황인지 한 번씩 생각해보자