[Algorithm] [Python&Swift] Programmers - 모의고사

모의고사

---

문제 설명

---

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

제출 코드

---

Python

def solution(answers):     
   answer = []
   cnt = [0, 0, 0]
   ans_one = [1,2,3,4,5] * 2000
   ans_two = [2,1,2,3,2,4,2,5] * 1250
   ans_thr = [3,3,1,1,2,2,4,4,5,5] * 1000

   for flag, one, two, thr in zip(answers, ans_one, ans_two, ans_thr):
       if flag == one:
           cnt[0] += 1
       if flag == two:
           cnt[1] += 1
       if flag == thr:
           cnt[2] += 1

   for i in range(len(cnt)):
       if cnt[i] == max(cnt):
           answer.append(i+1)
   return answer

코드 설명

---

Line 4~6 : 수포자 3명이 답을 찍는 패턴을 answers최대 길이에 맞춰 작성한다.

Line 8~14 : zip을 통해 정답, 수포자1의 답, 수포자2의 답, 수포자3의 답을 순차적으로 가져와 각각 비교하며
정답일 경우 cnt배열의 정답자 idx에 1을 더한다.

Line 16~18 : cnt list를 순회하며 최대값 즉, 제일 많이 맞춘 사람을 찾고 그 사람의 idx+1 값을 answer 리스트에 덧붙힌다.

정답은 맞았지만, answer의 길이가 10000이 아닐경우 answer_one과 answer_two, answer_thr 모두 길이 10000으로 generate하기 때문에 위 코드는 매우 매우 최악의 효율이다..

다른 사람의 풀이

---

from itertools import cycle

def solution(answers):
   giveups = [
       cycle([1,2,3,4,5]),
       cycle([2,1,2,3,2,4,2,5]),
       cycle([3,3,1,1,2,2,4,4,5,5]),
   ]
   scores = [0, 0, 0]
   for num in answers:
       for i in range(3):
           if next(giveups[i]) == num:
               scores[i] += 1
   highest = max(scores)

   return [i + 1 for i, v in enumerate(scores) if v == highest]

위 코드는 내가 실수한 부분에 대하여 완벽히 보완되어진 코드이다.

수포자들의 정답 패턴을 itertools의 cycle을 활용하여 generator로 처리하여 공간복잡도까지 고려하였다.

다른 사람의 풀이 2

---

def solution(answers):
   pattern1 = [1,2,3,4,5]
   pattern2 = [2,1,2,3,2,4,2,5]
   pattern3 = [3,3,1,1,2,2,4,4,5,5]
   score = [0, 0, 0]
   result = []

   for idx, answer in enumerate(answers):
       if answer == pattern1[idx%len(pattern1)]:
           score[0] += 1
       if answer == pattern2[idx%len(pattern2)]:
           score[1] += 1
       if answer == pattern3[idx%len(pattern3)]:
           score[2] += 1

   for idx, s in enumerate(score):
       if s == max(score):
           result.append(idx+1)

   return result

위 코드는 enumerate를 활용하여 코드를 작성하였다.

또한, 패턴의 길이는 패턴의 순환주기에 따라 각각 게산해주는 방식을 사용하였다.

정답 검증 로직은 enumerate를 이용해 answers 리스트에서 answer를 , pattern 리스트에서 인덱싱을 통해

각각의 값들을 순차적으로 검증한다.

매우 깔끔한 코드다!

 

Swift

func solution(_ answers: [Int]) -> [Int] {
    // answers: 정답 배열
    var point: [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 totalScore: [Int] = [0, 0, 0]

    for (idx, value) in answers.enumerated() {
        if value == answer.a[idx % 5] { totalScore[0] += 1 }
        if value == answer.b[idx % 8] { totalScore[1] += 1 }
        if value == answer.c[idx % 10] { totalScore[2] += 1 }
    }

    guard let maxPoint = totalScore.max() else {
        return []
    }

    for (i, v) in totalScore.enumerated() {
        if v >= maxPoint { point.append(i+1) }
    }

    return point.sorted(by: <)
}