[프로그래머스-Lv1] 모의고사 / Python

코딩테스트 연습 - 모의고사

import sys

input = sys.stdin.readline
answers = list(map(int, input().split()))


def solution(answers):
    first = [1, 2, 3, 4, 5]
    second = [2, 1, 2, 3, 2, 4, 2, 5]
    third = [3, 3, 1, 1, 2, 2, 4, 4, 5, 5]
    score = [0, 0, 0]
    for i in range(len(answers)):
        if answers[i] == first[i % 5]:
            score[0] += 1
        if answers[i] == second[i % 8]:
            score[1] += 1
        if answers[i] == third[i % 10]:
            score[2] += 1
    answer = []
    temp = max(score)
    for i in range(len(score)):
        if temp == score[i]:
            answer.append(i + 1)
    return answer


print(solution(answers))

 

해설

각 사람 별로 답을 맞힌 수를 1씩 증가시켜줬다.

마지막엔 가장 높은 값의 인덱스들만 추출해 리턴해줬다.