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

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

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

 

1️⃣ 수포자별로 찍는 순서를 리스트에 담아 주었고, 각각 for문을 돌려 정답 리스트와 맞는 개수를 cnts 배열에 더해주었다.

2️⃣ Max값을 max(cnts)로 두었고, for문을 돌려 Max와 같으면 answer 리스트에 삽입해주었다!

 

def solution(answers):
    answer = []
    cnts = [0]*3
    temp1 = [1,2,3,4,5]
    temp2 = [2,1,2,3,2,4,2,5]
    temp3 = [3,3,1,1,2,2,4,4,5,5]

    for idx in range(len(answers)):
        if temp1[idx%len(temp1)] == answers[idx]:
            cnts[0] += 1
    for idx in range(len(answers)):
        if temp2[idx%len(temp2)] == answers[idx]:
            cnts[1] += 1
    for idx in range(len(answers)):
        if temp3[idx%len(temp3)] == answers[idx]:
            cnts[2] += 1
    Max = max(cnts)
    for cnt in range(3):
        if cnts[cnt] == Max:
            answer.append(cnt+1)
    return answer