완전탐색 (Level 1) : 모의고사 (Python)

문제 : 

코딩테스트 연습 - 모의고사

 

Solution

def solution(answers):
    answer = []
    s1=[1,2,3,4,5]
    s2=[2,1,2,3,2,4,2,5]
    s3=[3,3,1,1,2,2,4,4,5,5]
    
    c1=c2=c3=0 #맞은 문제 수
    p1=p2=p3=0 #위치
    
    for i in range(len(answers)):
        if answers[i]==s1[p1]:c1+=1
        if answers[i]==s2[p2]:c2+=1
        if answers[i]==s3[p3]:c3+=1
        
        p1+=1
        p2+=1
        p3+=1
        
        if p1==len(s1):p1=0
        if p2==len(s2):p2=0
        if p3==len(s3):p3=0
    
    maxscore=max(c1,c2,c3)
    if c1==maxscore:answer.append(1)
    if c2==maxscore:answer.append(2)
    if c3==maxscore:answer.append(3)
    
    return answer