[Programmers] 모의고사 with Python

문제

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

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

풀이

각각의 수포자들 패턴과 답안을 비교해서 카운트

enumerate() 함수를 이용해 인덱스 값을 받아 출력한다.

답

def solution(answers): 
    no_1 = [1, 2, 3, 4, 5]
    no_2 = [2, 1, 2, 3, 2, 4, 2, 5]
    no_3 = [3, 3, 1, 1, 2, 2, 4, 4, 5, 5]
    cnt = [0, 0, 0]

    for i in range(len(answers)) :
        if no_1[i % len(no_1)] == answers[i] : # 순환주기
            cnt[0] += 1
        if no_2[i % len(no_2)] == answers[i] :
            cnt[1] += 1
        if no_3[i % len(no_3)] == answers[i] :
            cnt[2] += 1
    return [i+1 for i, j in enumerate(cnt) if j == max(cnt)]