프로그래머스. 완전탐색 문제 모의고사 파이썬 풀이

프로그래머스 완전탐색 level 1 모의고사 https://programmers.co.kr/learn/courses/30/lessons/42840

파이썬 풀이

def solution(answers):
    answer = []

    array_1 = [1, 2, 3, 4, 5]
    array_2 = [2, 1, 2, 3, 2, 4, 2, 5]
    array_3 = [3, 3, 1, 1, 2, 2, 4, 4, 5, 5]

    count = [0, 0, 0]

    for i in range(len(answers)):
        if array_1[i % len(array_1)] == answers[i]:
            count[0] += 1
        if array_2[i % len(array_2)] == answers[i]:
            count[1] += 1
        if array_3[i % len(array_3)] == answers[i]:
            count[2] += 1

    max_count = max(count)

    for i in range(len(count)):
        if max_count == count[i]:
            answer.append(i + 1)

    return answer

최대값을 결과에 삽입하는 과정을 enumerate를 활용하여 아래와 같이 바꿀 수 있다.

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