[LV.3] 완전탐색 > 모의고사 ⭐️

문제

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

프로그래머스

---

문제 설명

• 수포자(수학을 포기한 사람) 삼인방은 모의고사에 수학 문제를 전부 찍으려는 상황
• 수포자는 1번 문제부터 마지막 문제까지 다음과 같이 찍음

• 1번 문제부터 마지막 문제까지의 정답이 순서대로 들은 배열 answers가 주어졌을 때, 가장 많은 문제를 맞힌 사람이 누구인지 배열에 담아 반환하는 함수 작성
• 제한 조건
  시험은 최대 10,000 문제로 구성
  문제의 정답은 1, 2, 3, 4, 5중 하나
  가장 높은 점수를 받은 사람이 여럿일 경우, return하는 값을 오름차순 정렬

---

정답 코드

def solution(answers):
    # 수포자 1, 2, 3의 찍기 패턴 정의
    first = [1, 2, 3, 4, 5]
    second = [2, 1, 2, 3, 2, 4, 2, 5]
    third = [3, 3, 1, 1, 2, 2, 4, 4, 5, 5]

    # 각 수포자의 점수를 저장할 리스트 (초기값 0)
    score = [0, 0, 0]

    # 정답과 수포자들의 답을 하나씩 비교하면서 점수 계산
    for i, ans in enumerate(answers):
        # 수포자 1의 패턴과 정답 비교
        if first[i % len(first)] == ans:
            score[0] += 1
        # 수포자 2의 패턴과 정답 비교
        if second[i % len(second)] == ans:
            score[1] += 1
        # 수포자 3의 패턴과 정답 비교
        if third[i % len(third)] == ans:
            score[2] += 1

    # 세 수포자 중 가장 높은 점수 계산
    max_score = max(score)

    # 최고 점수를 받은 수포자의 번호(1, 2, 3 중 해당하는 것)를 리스트로 저장
    result = [i + 1 for i, s in enumerate(score) if s == max_score]
    # → i + 1을 하는 이유: 수포자 번호는 1번부터 시작이기 때문 (파이썬 인덱스는 0부터)

    # 결과 반환
    return result

---

다른 정답 코드 (확장성 향상 - 패턴과 점수 묶어서 관리)

- 새로운 수포자가 생겨도 반복문을 통해 쉽게 처리 가능

def solution(answers):
    # 수포자 1, 2, 3의 찍기 패턴을 리스트로 저장
    patterns = [
        [1, 2, 3, 4, 5],                # 수포자 1
        [2, 1, 2, 3, 2, 4, 2, 5],       # 수포자 2
        [3, 3, 1, 1, 2, 2, 4, 4, 5, 5]  # 수포자 3
    ]

    # 각 수포자의 점수를 저장할 리스트 (초기값은 모두 0)
    scores = [0] * len(patterns)       # [0, 0, 0]

    # 문제 번호(i)와 해당 문제의 정답(ans)을 순회
    for i, ans in enumerate(answers):
        # 각 수포자의 패턴을 순회하며 정답과 비교
        for idx, pattern in enumerate(patterns):
            # 수포자의 패턴은 반복되므로, 인덱스를 나머지 연산으로 맞춰줌
            if pattern[i % len(pattern)] == ans:
                # 정답과 일치하면 해당 수포자의 점수 1 증가
                scores[idx] += 1

    # 세 수포자 중 가장 높은 점수를 구함
    max_score = max(scores)

    # 최고 점수를 받은 수포자의 번호(1번부터 시작)를 리스트로 반환
    result = [i + 1 for i, s in enumerate(scores) if s == max_score]

    # 결과 반환
    return result

---

첫 번째 시도 (실패)

def solution(answers):
    first = [1, 2, 3, 4, 5]
    second = [2, 1, 2, 3, 2, 4, 2, 5]
    third = [3, 3, 1, 1, 2, 2, 4, 4, 5, 5]

    score = [0, 0, 0]

    for i, ans in enumerate(answers):

 

• 각 문제에 대해 각 수포자가 찍은 답과 실제 정답을 비교하려면?
  • 수포자들은 일정한 패턴대로 정답을 찍는 상황
  • 문제는 최대 10,000개이기 때문에 패턴을 무한히 반복하는 것처럼 적용 필요
• second = [2, 1, 2, 3, 2, 4, 2, 5] 의 경우
  • 인덱스 0~7까지만 접근 가능
  • 문제 번호가 커질수록, 패턴 인덱스를 이렇게 반복해야 됨 (0 1 2 3 4 5 6 7 0 1 2 3 ...) → i % len(second)

---

두 번째 시도 (성공)

def solution(answers):
    first = [1, 2, 3, 4, 5]
    second = [2, 1, 2, 3, 2, 4, 2, 5]
    third = [3, 3, 1, 1, 2, 2, 4, 4, 5, 5]

    score = [0, 0, 0]

    for i, ans in enumerate(answers):
        if first[i%len(first)] == ans:
            score[0] += 1
        if second[i%len(second)] == ans:
            score[1] += 1
        if third[i%len(third)] == ans:
            score[2] += 1
    
    max_score = max(score)
    
    result = [i+1 for i, s in enumerate(score) if s == max_score]
    return result