[Python]프로그래머스 - 모의고사(level1)

문제출처 - https://programmers.co.kr/learn/courses/30/lessons/42840

코딩테스트 연습 - 모의고사 | 프로그래머스

def solution(answers):
    answer = []
    one = [1,2,3,4,5]
    two = [2,1,2,3,2,4,2,5]
    three = [3,3,1,1,2,2,4,4,5,5]
    score = [0,0,0]
    
    for i in range(len(answers)):
        if answers[i] == one[i % len(one)]: 
            score[0] += 1
        if answers[i] == two[i % len(two)]: 
            score[1] += 1
        if answers[i] == three[i % len(three)]: 
            score[2] += 1
    
    max_score = max(score)
    for i in range(3):
        if score[i] == max_score:
            answer.append(i+1)
    
    return answer