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

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

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

프로그래머스

 

내 코드

def solution(answers):
    tmp_answer = [0]*3
    way_1 = [1,2,3,4,5]
    way_2 = [2,1,2,3,2,4,2,5]
    way_3 = [3,3,1,1,2,2,4,4,5,5]
    answer = []
    for i in range(len(answers)):
        if answers[i] == way_1[i%(len(way_1))]:
            tmp_answer[0] += 1
        if answers[i] == way_2[i%(len(way_2))]:
            tmp_answer[1] += 1
        if answers[i] == way_3[i%(len(way_3))]:
            tmp_answer[2] += 1
    
    for i in range(3):
        if tmp_answer[i] == max(tmp_answer):
            answer.append(i+1)
    return answer

찾아보고 여기에는 쓰지않았지만 max값의 인덱스값을 알고싶다면 list.index(max(list)를 하면된다. 다만 max값을 가진 인덱스가 여러개여도 가장 앞의 것 하나만 반환해준다.