프로그래머스 모의고사 (level 1)

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

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

def solution(answers):
    answer = []
    a = [0, 0, 0]
    p1 = [1,2,3,4,5]
    p2 = [2,1,2,3,2,4,2,5]
    p3 = [3,3,1,1,2,2,4,4,5,5]
    pos1, pos2, pos3 = 0, 0, 0
    for i in range(len(answers)):
        if answers[i] == p1[pos1]:
            a[0] += 1
            
        if answers[i] == p2[pos2]:
            a[1] += 1 
            
        if answers[i] == p3[pos3]:
            a[2] += 1     
            
        pos1 += 1
        pos2 += 1
        pos3 += 1
        if pos1 == 5:
            pos1 = 0
        if pos2 == 8:
            pos2 = 0
        if pos3 == 10:
            pos3 = 0
            
    for i in range(3):
        if a[i] == max(a):
            answer.append(i+1)
    
    return answer