프로그래머스 : 모의고사(Python)

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

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

1번 사람부터 3번 사람까지 찍는 답은 규칙적이다.

1번 사람 : [1,2,3,4,5] 반복

2번 사람 : [2,1,2,3,2,4,2,5] 반복

3번 사람 : [3,3,1,1,2,2,4,4,5,5] 반복

 

그러므로 답안과 사람들의 답안을 비교하며 count 숫자를 늘리면 된다. 비교를 할 때, 사람들의 답이 반복되므로 정답 리스트의 길이를 사람들의 답 길이로 나눈 나머지를 이용하여 비교하면 된다. 

 

def solution(answers):
    answer = []
    arr1 = [1,2,3,4,5]
    arr2 = [2,1,2,3,2,4,2,5]
    arr3 = [3,3,1,1,2,2,4,4,5,5]
    arr1_count = 0
    arr2_count = 0
    arr3_count = 0
    reuslt = []
    
    for i in range(len(answers)):
        if answers[i] == arr1[i % len(arr1)]:
            arr1_count += 1
        if answers[i] == arr2[i % len(arr2)]:
            arr2_count += 1
        if answers[i] == arr3[i % len(arr3)]:
            arr3_count += 1
    
    result = [arr1_count,arr2_count,arr3_count]
    
    for i,j in enumerate(result):
        if j == max(result):
            answer.append(i+1)
    
    return answer