[프로그래머스/파이썬] 모의고사

-코드

def solution(answers):
    answer = []
    p1 = [1, 2, 3, 4, 5]
    p2 = [2, 1, 2, 3, 2, 4, 2, 5]
    p3 = [3, 3, 1, 1, 2, 2, 4, 4, 5, 5]
    r = [0, 0, 0]
    for i in range(len(answers)):
        if p1[i % 5] == answers[i]:
            r[0] += 1
        if p2[i % 8] == answers[i]:
            r[1] += 1
        if p3[i % 10] == answers[i]:
            r[2] += 1
    for i in range(3):
        if r[i] == max(r):
            answer.append(i + 1)
    return answer