[Python] 모의고사

hyeon·2021년 2월 15일
0

Programmers

목록 보기
10/18
post-thumbnail

def solution(answers):
    answer = []
    one = [1,2,3,4,5]
    two = [2,1,2,3,2,4,2,5]
    three = [3,3,1,1,2,2,4,4,5,5]
    
    score = [0,0,0]
    
    for i, a in enumerate(answers) :
        if(a == one[i%(len(one))]) : score[0] +=1
        if(a == two[i%(len(two))]) : score[1] +=1
        if(a == three[i%(len(three))]) : score[2] +=1
    
    m = max(score)
    answer = [i+1 for i, s in enumerate(score) if m == s]
    
    return answer
    
profile
바스락바스락

0개의 댓글

관련 채용 정보