[프로그래머스/1단계/파이썬(Python3)] 모의고사

문제 출처

 

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

 

풀이 코드

def solution(answers):
    one_ans = [1, 2, 3, 4, 5]
    two_ans = [2, 1, 2, 3, 2, 4, 2, 5]
    thr_ans = [3, 3, 1, 1, 2, 2, 4, 4, 5, 5]
    
    n=len(answers)
    score=[0,0,0]
    
    one_ans=one_ans*(n//5)+one_ans
    two_ans=two_ans*(n//8)+two_ans
    thr_ans=thr_ans*(n//10)+thr_ans
    
    for i in range(len(answers)):
        if one_ans[i] == answers[i]: score[0]+=1
        if two_ans[i] == answers[i]: score[1]+=1
        if thr_ans[i] == answers[i]: score[2]+=1
    
    answer = [i+1 for i, value in enumerate(score) if value==max(score)]
    
    return answer