[프로그래머스] 20. 모의고사 - Python

프로그래머스 - 모의고사

 

문제집 링크 : http://aladin.kr/p/szPra

내일은 코딩테스트 with 파이썬

 

프로그래머스 URL :  

[도서실습] 내일은 코딩테스트 with 파이썬(자료구조와 알고리즘의 기초부터 실전까지)

---

■ 작성 코드

def solution(answers):
    student1 = [1,2,3,4,5]
    student2 = [2,1,2,3,2,4,2,5]
    student3 = [3,3,1,1,2,2,4,4,5,5]
    
    scores = [0,0,0]
    ans = []
    
    for idx, answer in enumerate(answers):
    	if answer == student1[idx%len(student1)]:
        	scores[0] += 1
    	if answer == student2[idx%len(student2)]:
        	scores[1] += 1
    	if answer == student3[idx%len(student3)]:
        	scores[2] += 1   
            
    for idx, s in enumerate(scores):
    	if s == max(scores):
        	ans.append(idx+1)
            
    return ans