[프로그래머스] 완전 탐색 문제 - 모의 고사

문제 : 모든 문제를 찍는 사람의 패턴 + 문제 정답이 주어질 때 제일 점수가 잘 나온 사람은?

 

def solution(answers):
    answer = []
    score = [0,0,0]
    students = [[1,2,3,4,5],[2, 1, 2, 3, 2, 4, 2, 5],[3, 3, 1, 1, 2, 2, 4, 4, 5, 5]]
    i = 0
    for a in answers:
        if a == students[0][i % 5]:
            score[0] += 1
        if a == students[1][i % 8]:
            score[1] += 1
        if a == students[2][i % 10]:
            score[2] += 1
        i += 1
    print(score)
    answer = [i+1 for i in range(3) if score[i] == max(score)]
    return answer

쉽고 재밌는 문제