23/12/07

프로그래머스 모의고사

문제

 

def solution(answers):
    answer = []
    a = [1, 2, 3, 4, 5]
    b = [2, 1, 2, 3, 2, 4, 2, 5]
    c = [3, 3, 1, 1, 2, 2, 4, 4, 5, 5]

    count=[0,0,0]
    
    for idx,value in enumerate(answers):
        if a[idx % len(a)] == value:
            count[0]+=1
        if b[idx % len(b)] == value:
            count[1]+=1
        if c[idx % len(c)] == value:
            count[2]+=1   

    rank_max= max(count)
    
    for idx,value in enumerate(count):
        if value == rank_max:
            answer.append(idx+1)
    
    return answer

 

1~3번 수포자 각각 패턴을 a~c 리스트에 넣고 enumerate로 answers를 돌리면서

idx에서 각 a~c리스트의 길이로 나눈 나머지 값으로 각 a~c의 나머지값 인덱스로 value랑 맞나 체크하고 

a~c 순위를 결정하게 한다.