[프로그래머스 | 파이썬 python] 모의고사

🌏 문제

프로그래머스

 

1️⃣ 풀이

def solution(answers):
    stu1 = [1, 2, 3, 4, 5]
    stu2 = [2, 1, 2, 3, 2, 4, 2, 5]
    stu3 = [ 3, 3, 1, 1, 2, 2, 4, 4, 5, 5]
    
    correct=[0,0,0]
    for idx, answer in enumerate(answers):
        if stu1[idx%len(stu1)] == answer:
            correct[0] += 1
        if stu2[idx%len(stu2)] == answer:
            correct[1] += 1
        if stu3[idx%len(stu3)] == answer:
            correct[2] += 1
            
    return [idx+1 for idx, x in enumerate(correct) if x == max(correct)]