프로그래머스 모의고사
문제 설명
수포자는 수학을 포기한 사람의 준말입니다. 수포자 삼인방은 모의고사에 수학 문제를 전부 찍으려 합니다. 수포자는 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하는 값을 오름차순 정렬해주세요.
[입출력 예]
| answers | return |
|---|---|
| [1,2,3,4,5] | [1] |
| [1,3,2,4,2] | [1,2,3] |
문제 해석하기
완전 탐색 문제이기 때문에, 수포자 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,
⇒ 각각의 규칙을 배열에 넣어주고, answer의 길이에 따라 계속 반복해주도록 한다.
- 처음에는 itertools.cycle을 이용해서 해결해보려고 했었는데, 아래의 코드처럼
dropout2[i%8]나머지를 이용해서 반복하는 방법으로 재구현하였다. - score배열에 몇개를 맞혔는지 넣어주고, 해당 값을 돌리면서 가장 많이 맞힌 사람을 찾아주었다.
- 위의 과정에서 enumerate함수를 사용해서 이름과 수를 모두 받아줄 수 있도록 했다.
def solution(answers):
dropout1 = [1,2,3,4,5]
dropout2 = [2,1,2,3,2,4,2,5]
dropout3 = [3,3,1,1,2,2,4,4,5,5]
score=[0,0,0]
for i in range(len(answers)):
if answers[i] == dropout1[i%5]:
score[0] += 1
if answers[i] == dropout2[i%8]:
score[1] += 1
if answers[i] == dropout3[i%10]:
score[2] += 1
for idx, num in enumerate(score):
if num == max(score):
answer.append(idx + 1)
return answerEnumerate 함수란?
시퀀스(리스트, 튜플)을 순회할 떄 인덱스와 값을 함께 반환하는 유용한 내장 함수임
- Python에서 제공하는 enumerate 내장 함수로 자료 구조의 인덱스와 데이터를 더 직관적이고 편리하게 활용이 가능하다.
- for문의 in 뒷부분을 enumerate()로 감싸주어 인덱스와 원소를 동시에 접근하면서 루프를 돌릴 수 있도록 하는 함수
오늘의 회고
enumerate라는 함수를 통해서 인덱스와 원소 모두 접근하면서 루프를 돌리는 방식에 대해 알 수 있어서 좋았다. 그리고, itertools를 사용해서 반복하려 했었지만, 더 쉽게 나머지를 이용해 반복시키는 방법을 찾아서 조금이나마 코드를 개선할 수 있었던 점이 재밌었다.
