본문 바로가기
프로그래머스

Level1 모의고사

by ysunee 2022. 2. 2.
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]
    a1=0;b1=0;c1=0
    n = len(answers)
    for i in range(n):
        if(a[i%5]==answers[i]): a1+=1
        if(b[i%8]==answers[i]): b1+=1    
        if(c[i%10]==answers[i]): c1+=1
    
    if(a1>b1 and a1>c1): answer.append(1)
    elif(b1>a1 and b1>c1): answer.append(2)
    elif(c1>a1 and c1>b1): answer.append(3)
    elif(a1==b1 and b1>c1): answer.append(1); answer.append(2)
    elif(a1==c1 and c1>b1): answer.append(1); answer.append(3)
    elif(c1==b1 and b1>a1): answer.append(2); answer.append(3)
    elif(a1==b1 and b1==c1): answer.append(1); answer.append(2); answer.append(3)
        
    return answer

'프로그래머스' 카테고리의 다른 글

Level1 폰켓몬  (0) 2022.02.02
Level1 체육복  (0) 2022.02.02
Level1 K번째 수  (0) 2022.02.02
Level1 소수 만들기  (0) 2022.02.02
Level1 내적  (0) 2022.02.02

댓글