프로그래머스 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): s0 = [1, 2, 3, 4, 5] s1 = [2, 1, 2, 3, 2, 4, 2, 5] s2 = [3, 3, 1, 1, 2, 2, 4, 4, 5, 5] answer = [0, 0, 0] for i in range(len(answers)): if answers[i] == s0[i % len(s0) if i != 0 else 0]: answer[0] += 1 if answers[i] == s1[i % len(s1) if i != 0 else 0]: answer[1] += 1 if answers[i] == s2[i % len(s2) if i != 0 else 0]: answer[2] += 1 return [i+1 for i in range(len(answer)) if answer[i] == max(answer)]
다른사람 풀이
from itertools import cycle def solution(answers): s = [ cycle([1,2,3,4,5]), cycle([2,1,2,3,2,4,2,5]), cycle([3,3,1,1,2,2,4,4,5,5]), ] scores = [0, 0, 0] for num in answers: for i in range(3): if next(s[i]) == num: scores[i] += 1 highest = max(scores) return [i + 1 for i, v in enumerate(scores) if v == highest]
배운 점
⦁ 무한 반복자 - itertools.cycle
Author And Source
이 문제에 관하여(프로그래머스 LV1 - 모의고사), 우리는 이곳에서 더 많은 자료를 발견하고 링크를 클릭하여 보았다 https://velog.io/@minmin/프로그래머스-LV1-모의고사저자 귀속: 원작자 정보가 원작자 URL에 포함되어 있으며 저작권은 원작자 소유입니다.
우수한 개발자 콘텐츠 발견에 전념 (Collection and Share based on the CC Protocol.)