프로그래머스 모의고사

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

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

 

def solution(answers):
    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]
    a = [[1,0], [2,0], [3, 0]]
    
    for x in range(len(answers)):
        if p1[x % len(p1)] == answers[x]:
            a[0][1] += 1
        if p2[x % len(p2)] == answers[x]:
            a[1][1] += 1
        if p3[x % len(p3)] == answers[x]:
            a[2][1] += 1
    
    a.sort(key = lambda x: x[1], reverse=True)
    return [x[0] for x in a if a[0][1] == x[1]]