문제

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

코드

def solution(answers):
    answer = []
    
    solved = [0] * 3
    one = [1, 2, 3, 4, 5] # 5
    two = [2, 1, 2, 3, 2, 4, 2, 5] # 8
    three = [3, 3, 1, 1, 2, 2, 4, 4, 5, 5] # 10
    
    # 1번 친구 정답 개수 체크
    for i in range(len(answers)):
        if answers[i] == one[i%5]:
            solved[0] += 1
    # 2번 친구 정답 개수 체크
    for i in range(len(answers)):
        if answers[i] == two[i%8]:
            solved[1] += 1
    # 3번 친구 정답 개수 체크
    for i in range(len(answers)):
        if answers[i] == three[i%10]:
            solved[2] += 1
            
    # 많이 맞춘 친구는?
    mx = max(solved)
    for i in range(0, 3):
        if solved[i] == mx:
            answer.append(i+1)
    
    return answer