문제

모의고사

문제 설명

수포자는 수학을 포기한 사람의 준말입니다.
수포자 삼인방은 모의고사에 수학 문제를 전부 찍으려 합니다.
수포자는 1번 문제부터 마지막 문제까지 다음과 같이 찍습니다.

• 1번 수포자가 찍는 방식: 1, 2, 3, 4, 5, 1, 2, 3, 4, 5, $\cdots$
• 2번 수포자가 찍는 방식: 2, 1, 2, 3, 2, 4, 2, 5, 2, 1, 2, 3, 2, 4, 2, 5, $\cdots$
• 3번 수포자가 찍는 방식: 3, 3, 1, 1, 2, 2, 4, 4, 5, 5, 3, 3, 1, 1, 2, 2, 4, 4, 5, 5, $\cdots$

1번 문제부터 마지막 문제까지의 정답이 순서대로 들은 배열 answers가 주어졌을 때, 가장 많은 문제를 맞힌 사람이 누구인지 배열에 담아 return 하도록 solution 함수를 작성해주세요.

---

제한사항

• 시험은 최대 10,000 문제로 구성되어 있습니다.
• 문제의 정답은 1, 2, 3, 4, 5 중 하나입니다.
• 가장 높은 점수를 받은 사람이 여럿일 경우, return하는 값을 오름차순으로 정렬해주세요.

---

입출력 예시

answers	result
[1,2,3,4,5]	[1]
[1,3,2,4,2]	[1,2,3]

---

문제 풀이

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 반복

정답: 1, 2, 3, 4, 5

1: 1, 2, 3, 4, 5
2: 2, 1, 2, 3, 2
3: 3, 3, 1, 1, 2

1: O, O, O, O, O $\rightarrow$ 5
2: X, X, X, X, X $\rightarrow$ 0
3: X, X, X, X, X $\rightarrow$ 0

정답: 1, 3, 2, 4, 2

1: O, X, X, O, X $\rightarrow$ 2
2: X, X, O, X, O $\rightarrow$ 2
3: X, O, X, X, O $\rightarrow$ 2

문제가 최대 10,000개이므로 수포자 1의 경우 5개(1, 2, 3, 4, 5)를 반복하면서 찍어야 하고, 수포자 2의 경우 8개를 반복하면서 찍어야 하고, 수포자 3의 경우 10개를 반복하면서 찍어야 한다.

찍는 번호와 문제의 번호를 모듈러 함수를 통해 인덱스로 반복해줘야 한다.

문제의 번호 % 가지고 있는 찍는 방식의 개수

예를 들어 문제가 10개라고 하면 인덱스는 [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]이다.
수포자 1의 찍는 방식의 개수는 5개이다.

0 % 5 = 0
1 % 5 = 1
2 % 5 = 2
3 % 5 = 3
4 % 5 = 4
5 % 5 = 0
6 % 5 = 1
7 % 5 = 2
8 % 5 = 3
9 % 5 = 4

다음은 인덱스를 기반으로 문제의 답과 찍은 번호가 일치하면 정답수 증가한다.

마지막으로 정답수가 가장 높은 사람을 오름차순으로 정렬해서 반환한다.

최초 문제 풀이

def solution(answers):
    first_options = [1, 2, 3, 4, 5]
    second_options = [2, 1, 2, 3, 2, 4, 2, 5]
    third_options = [3, 3, 1, 1, 2, 2, 4, 4, 5, 5]

    first_answers = 0
    second_answers = 0
    third_answers = 0

    winner = []

    for i in range(len(answers)):
        first_index = i % len(first_options)
        second_index = i % len(second_options)
        third_index = i % len(third_options)

        if answers[i] == first_options[first_index]:
            first_answers += 1

        if answers[i] == second_options[second_index]:
            second_answers += 1

        if answers[i] == third_options[third_index]:
            third_answers += 1

    max_count = max(first_answers, second_answers, third_answers)

    if max_count == 0:
        winner.append(1)
        winner.append(2)
        winner.append(3)
    else:
        if max_count == first_answers:
            winner.append(1)
        if max_count == second_answers:
            winner.append(2)
        if max_count == third_answers:
            winner.append(3)

    return sorted(winner)

리팩터링을 한 번 해보자.

찍는 방식은 하나의 배열로 만들어보자.

methods = [
	[1, 2, 3, 4, 5],
    [2, 1, 2, 3, 2, 4, 2, 5],
    [3, 3, 1, 1, 2, 2, 4, 4, 5, 5]
]

찍는 방식과 마찬가지로 정답 횟수도 하나의 배열로 만들어보자.

scores = [0] * 3

answers는 인덱스를 사용할 필요 없이 enumerate를 사용하자.

for i, answer in enumerate(answers):
	first_index = i % len(methods[0])
    second_index = i % len(methods[1])
    third_index = i % len(methods[2])

    if answer == methods[0][first_index]:
        scores[0] += 1

    if answer == methods[1][second_index]:
        scores[1] += 1

    if answer == methods[2][third_index]:
        scores[2] += 1

최종 문제 풀이

def solution(answers):
    methods = [
        [1, 2, 3, 4, 5],
        [2, 1, 2, 3, 2, 4, 2, 5],
        [3, 3, 1, 1, 2, 2, 4, 4, 5, 5]
    ]

    scores = [0] * 3

    winner = []

    for i, answer in enumerate(answers):
        first_index = i % len(methods[0])
        second_index = i % len(methods[1])
        third_index = i % len(methods[2])

        if answer == methods[0][first_index]:
            scores[0] += 1

        if answer == methods[1][second_index]:
            scores[1] += 1

        if answer == methods[2][third_index]:
            scores[2] += 1

    max_count = max(scores)

    for i, score in enumerate(scores):
        if score == max_count:
            winner.append(i + 1)

    return winner

---

느낀점

enumerate

파이썬에서 리스트(또는, 튜플, 문자열 등)를 순회할 때 인덱스와 원소의 값을 튜플 형태로 동시에 반환해주는 내장 함수이다.

기존 방식 (range와 len 사용)

fruits = ['apple', 'banana', 'orange']

for i in range(len(fruits)):
    print(f"인덱스: {i}, 값: {fruits[i]}")

# 인덱스: 0, 값: apple
# 인덱스: 1, 값: banana
# 인덱스: 2, 값: orange

enumarate() 활용

fruits = ['apple', 'banana', 'orange']

for i, fruit in enumerate(fruits):
    print(f"인덱스: {i}, 값: {fruit}")

# 인덱스: 0, 값: apple
# 인덱스: 1, 값: banana
# 인덱스: 2, 값: orange