[프로그래머스/파이썬] 완전탐색 모의고사

https://programmers.co.kr/learn/courses/30/lessons/42840

---

알고리즘 분류

• 브루트포스

문제풀이

correct리스트는 각각의 수포자가 맞춘 문제 수를 저장한다.

해당 문제 수와 수포자를 함께 저장하고 문제를 많이 맞힌 순서대로 내림차순 정렬한다.

그리고 해당하는 수포자(인덱스)를 출력해준다.

소스코드

def solution(answers):
    stu1=[1,2,3,4,5]*2000
    stu2=[2,1,2,3,2,4,2,5]*1250
    stu3=[3,3,1,1,2,2,4,4,5,5]*1000

    correct=[0,0,0]
    for i in range(len(answers)):
        if answers[i]==stu1[i]:
            correct[0]+=1
        if answers[i]==stu2[i]:
            correct[1]+=1
        if answers[i]==stu3[i]:
            correct[2]+=1

    array=[]
    for i in range(3):
        array.append((correct[i],i+1))

    array.sort(key=lambda x: (-x[0]))

    result=[]
    temp=0
    for i in array:
        value,index=i
        if value>=temp:
            result.append(index)
            temp=value
    return result