[프로그래머스 고득점 kit] 완전탐색 #1_모의고사

( 문제 설명 )

수포자는 수학을 포기한 사람의 준말입니다. 수포자 삼인방은 모의고사에 수학 문제를 전부 찍으려 합니다. 수포자는 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하는 값을 오름차순 정렬해주세요.

입출력 예answersreturn

[1,2,3,4,5]	[1]
[1,3,2,4,2]	[1,2,3]

입출력 예 설명

입출력 예 #1

• 수포자 1은 모든 문제를 맞혔습니다.
• 수포자 2는 모든 문제를 틀렸습니다.
• 수포자 3은 모든 문제를 틀렸습니다.

따라서 가장 문제를 많이 맞힌 사람은 수포자 1입니다.

입출력 예 #2

• 모든 사람이 2문제씩을 맞췄습니다.

---

( 문제 해설 )

 

# 생각 알고리즘

1. 1번, 2번, 3번 수포자의 찍는 규칙은 정해져 있으니 리스트로 뽑아 옵니다.

_1 = [1, 2, 3, 4, 5]
_2 = [2, 1, 2, 3, 2, 4, 2, 5]
_3 = [3, 3, 1, 1, 2, 2, 4, 4, 5, 5]

2. answer 리스트를 순회하면 각각 1번, 2번, 3번 수포자들의 정답과 같은지 체크하면서 맞는 사람에게 +1을 해줍니다

2-1. answer리스트는 수포자의 규칙보다 많을 수 있기 때문에 그것을 고려하여 모듈러 연산을 이용하여 배열의 초과를 방지합니다.

if _1[i % len(_1)] == answers[i]:
            a[1] += 1
        if _2[i % len(_2)] == answers[i]:
            b[1] += 1
        if _3[i % len(_3)] == answers[i]:
            c[1] += 1

3-1. 많은 순으로 정렬한 다음 출력해주면 됩니다.

 

---

[ 코드 ]

 

def solution(answers):
    answer = []
    _1 = [1, 2, 3, 4, 5]
    _2 = [2, 1, 2, 3, 2, 4, 2, 5]
    _3 = [3, 3, 1, 1, 2, 2, 4, 4, 5, 5]
    a = [1, 0]
    b = [2, 0]
    c = [3, 0]

    for i in range(len(answers)):
        if _1[i % len(_1)] == answers[i]:
            a[1] += 1
        if _2[i % len(_2)] == answers[i]:
            b[1] += 1
        if _3[i % len(_3)] == answers[i]:
            c[1] += 1

    answer.append(a)
    answer.append(b)
    answer.append(c)
    
    res =[]
    answer.sort(key= lambda x: -x[1])
    for i in range(len(answer)):
        if answer[0][1] == answer[i][1]:
            res.append(answer[i][0])
   

    return res

 

---

후기

 

오랜만에 나오는 쉬워가기 문제

하지만 문제는 쉽지만 모듈러 연산의 유용성은 어디 분야든 많이 쓰이니

필요할 때 바로 생각나게끔 이해하고 넘어가는 건 필요합니다.