SW개발/코딩테스트

[프로그래머스]모의고사 - 완전탐색

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

 

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

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

programmers.co.kr

 

def solution(answers):
    # 수포자가 정답을 찍는 패턴을 미리 지정해둠
    person1 = [1, 2, 3, 4, 5]
    person2 = [2, 1, 2, 3, 2, 4, 2, 5]
    person3 = [3, 3, 1, 1, 2, 2, 4, 4, 5, 5]

    # 각 수포자 별로 맞춘 정답수를 리스트화
    score = [0, 0, 0]

    # enumerate를 통해 index, value 동시 이용
    for index, value in enumerate(answers):
        # index % (패턴의 길이)를 인덱스로 사용함으로써 패턴을 반복시킬 수 있음
        if person1[index % 5] == value:
            score[0] += 1
        if person2[index % 8] == value:
            score[1] += 1
        if person3[index % 10] == value:
            score[2] += 1

    answer = []
    for index, val in enumerate(score):
        # 가장 많이 맞춘 사람을 answer 리스트에 넣어줌
        # 이 때, 동일하게 맞춘 사람의 경우 오름차순으로 정렬되어 들어감
        if val ==  max(score):
            answer.append(index + 1)
    return answer

 

문제 해결 Point

먼저 완전탐색의 경우 가능한 경우의 수를 모두 반복시켜 정답을 도출해내는 방법이라고 간단하게 인지하고 문제를 풀이한다.

  1. 사람의 수는 3명으로 고정되어있고, 각 사람별로 패턴이 존재하므로 패턴을 리스트에 담아둔다.
  2. 정답 리스트를 enumerate로 순회하면서 index, value를 동시에 이용한다.
  3. 순회하면서 각 사람의 정답 패턴과 answer를 비교하는데, (index % 각 사람 패턴의 길이)를 통해서 패턴을 계속 반복시킬 수 있다.
  4. 더해진 값을 반복하면서 answer에 max 값이 되는 사람들만 리스트에 넣어준다. 
    (이 과정에서 자동적으로 오름차순 정렬이 되어 값이 들어간다.)

 

728x90