[Python] 모의 고사

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

프로그래머스

• 나의 풀이

# 모의고사

def solution(answers):
    supo_1 = 0
    supo_2 = 0
    supo_3 = 0
    
    supo_1_pattern = [1,2,3,4,5] * 8
    supo_2_pattern = [2,1,2,3,2,4,2,5] * 5
    supo_3_pattern = [3,3,1,1,2,2,4,4,5,5] * 4
    
    for i in range(len(answers)):
        idx = i % 40
        if supo_1_pattern[idx]==answers[i]:
            supo_1 += 1
        if supo_2_pattern[idx]==answers[i]:
            supo_2 += 1
        if supo_3_pattern[idx]==answers[i]:
            supo_3 += 1
            
    score = [supo_1, supo_2, supo_3]
    max_score = max(score)
    answer = []
    
    for i in range(3):
        if score[i] == max_score:
            answer.append(i+1)
    return sorted(answer)