모의고사

문제 설명

• 수포자는 수학을 포기한 사람의 준말입니다. 수포자 삼인방은 모의고사에 수학 문제를 전부 찍으려 합니다. 수포자는 1번 문제부터 마지막 문제까지 다음과 같이 찍습니다.

  1번 수포자가 찍는 방식: 1, 2, 3, 4, 5, 1, 2, 3, 4, 5, ...
  2번 수포자가 찍는 방식: 2, 1, 2, 3, 2, 4, 2, 5, 2, 1, 2, 3, 2, 4, 2, 5, ...
  3번 수포자가 찍는 방식: 3, 3, 1, 1, 2, 2, 4, 4, 5, 5, 3, 3, 1, 1, 2, 2, 4, 4, 5, 5, ...

• 1번 문제부터 마지막 문제까지의 정답이 순서대로 들은 배열 answers가 주어졌을 때, 가장 많은 문제를 맞힌 사람이 누구인지 배열에 담아 return 하도록 solution 함수를 작성해주세요.

제한 조건

• 시험은 최대 10,000 문제로 구성되어있습니다.
• 문제의 정답은 1, 2, 3, 4, 5중 하나입니다.
• 가장 높은 점수를 받은 사람이 여럿일 경우, return하는 값을 오름차순 정렬해주세요.

완전탐색

def solution(answers):
    answer = []
    # 수포자 1,2,3의 규칙 리스트
    sofoja1 = [1,2,3,4,5]
    sofoja2 = [2,1,2,3,2,4,2,5]
    sofoja3 = [3,3,1,1,2,2,4,4,5,5]
    # 수포자들이 각각 정답 수를 체크하기 위한 딕셔너리
    count = {1:0,2:0,3:0}
    
    # 반복문을 통한 완전 탐색
    for i in range(len(answers)):
    # 수포자들의 리스트 값과 정답 리스트의 값이 같을 때 체크
        if sofoja1[i%len(sofoja1)] == answers[i]:
            count[1] += 1
        if sofoja2[i%len(sofoja2)] == answers[i]:
            count[2] += 1
        if sofoja3[i%len(sofoja3)] == answers[i]:
            count[3] += 1
    # 딕셔너리의 value를 기준으로 내림차순으로 정렬
    result = sorted(count.items(),key=lambda x : x[1],reverse=True)
    # 가장 큰 값이 앞에 오므로 기준 값을 answer 리스트에 넣어줌
    answer.append(result[0][0])
	# 가장 큰 값과 같은 값을 가진 다른 요소가 있을 경우 answer 리스트에 추가
   
    for i in range(1,len(result)):
        if result[0][1] == result[i][1]:
            answer.append(result[i][0])
        else :
        	# 같지 않는 값을 만나면 반복문 중단
            break;
    
    #정답 반환
    return answer