[프로그래머스] lv.1 모의고사

문제

https://school.programmers.co.kr/learn/courses/30/lessons/42840

정답

def solution(answers):
    answer = []
    supo = [[1,2,3,4,5],[2, 1, 2, 3, 2, 4, 2, 5],[3, 3, 1, 1, 2, 2, 4, 4, 5, 5]]
    right = [0,0,0]
    
    for i in range(len(supo)):
        for j in range(len(answers)):
            n = len(supo[i])
            # print(i, j, n, j%n)
            if answers[j] == supo[i][j%n]:
                right[i] += 1
    for idx,score in enumerate(right):
        print(idx,score)
        if score == max(right):
            answer.append(idx+1)
    
    return answer

풀이

if answers[j] == supo[i][j%n]: 

다른 분의 풀이를 참고해서 해결했다.

 print(i, j, n, j%n)

supo를 answers 길이에 맞춰 순회할 수 있다.