[프로그래머스] Lv1. 모의고사

문제

수포자는 수학을 포기한 사람의 준말입니다. 수포자 삼인방은 모의고사에 수학 문제를 전부 찍으려 합니다. 수포자는 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 = []
    A = [1, 2, 3, 4, 5] * ((len(answers) // 5) + 1)
    B = [2, 1, 2, 3, 2, 4, 2, 5] * ((len(answers) // 8) + 1)
    C = [3, 3, 1, 1, 2, 2, 4, 4, 5, 5] * ((len(answers) // 10) + 1)
    count = [0, 0, 0]
    
    for i in range(0, len(answers)):
        if answers[i] == A[i]:
            count[0] += 1
        if answers[i] == B[i]:
            count[1] += 1
        if answers[i] == C[i]:
            count[2] += 1
    
    for i in range(1, 4):
        if count[i-1] == max(count):
            answer.append(i)
    
    return answer

제공되는 배열 answers의 길이 만큼 각 패턴을 반복 시켜주는 게 문제였다. 패턴이 더 길어지는 것은 문제가 없기 때문에 answers의 길이를 패턴의 길이로 나눈 값 +1을 패턴에 곱해준 뒤 인덱스 별로 값이 일치하는지를 비교했다.

값이 일치할 경우에는 count 안의 값을 1씩 올려주고, 최종적으로 count 안에서 max 값을 비교하여 가장 많은 문제를 맞힌 사람의 번호가 answer로 return 되도록 했다.

다른 사람들의 풀이:

def solution(answers):
    pattern1 = [1,2,3,4,5]
    pattern2 = [2,1,2,3,2,4,2,5]
    pattern3 = [3,3,1,1,2,2,4,4,5,5]
    score = [0, 0, 0]
    result = []

    for idx, answer in enumerate(answers):
        if answer == pattern1[idx%len(pattern1)]:
            score[0] += 1
        if answer == pattern2[idx%len(pattern2)]:
            score[1] += 1
        if answer == pattern3[idx%len(pattern3)]:
            score[2] += 1

    for idx, s in enumerate(score):
        if s == max(score):
            result.append(idx+1)

    return result

패턴 리스트를 별도로 늘려주지 않고도 사용할 패턴을 반복할 수 있도록 해주었다. 메모리를 훨씬 덜 차지하기 때문에 효율적일 듯하다.

비슷하지만 다른 방법으로 어떤 분은 cycle이라는 함수를 import 해와서 쓰는 경우도 있었다. 하지만 간단한 계산식으로 구현해낸 이 알고리즘이 훨씬 인상적이다.