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

일단 해볼게·2023년 12월 27일
0

프로그래머스

목록 보기
81/106

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

def solution(answers):
    num = [0, 0, 0] # 수포자가 맞춘 문제 수
    
    # 수포자 1, 2, 3
    math_fail1 = [1, 2, 3, 4, 5]
    math_fail2 = [2, 1, 2, 3, 2, 4, 2, 5]
    math_fail3 = [3, 3, 1, 1, 2, 2, 4, 4, 5, 5]
    
    for i in range(len(answers)):
        # 수포자가 찍는 방식의 값과 같은지 비교
        if answers[i] == math_fail1[i % len(math_fail1)]:
            num[0] += 1
        if answers[i] == math_fail2[i % len(math_fail2)]:
            num[1] += 1
        if answers[i] == math_fail3[i % len(math_fail3)]:
            num[2] += 1
    
    answer = []
    
    for i in range(3):
        if num[i] == max(num): # 최대값 answer에 추가
            answer.append(i + 1)
            
    return answer

수포자가 찍는 방식의 인덱스에 맞는 값과 answer[i]를 비교해서 count한다.

profile
시도하고 More Do하는 백엔드 개발자입니다.

0개의 댓글