프로그래머스 LEVEL1 - 모의고사

▶ 문제 설명

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

▶ 입출력 예

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

▶ 내가 짠 코드

def solution(answers):
    answer = []
    n1 = [1,2,3,4,5]
    n2 = [2,1,2,3,2,4,2,5]
    n3 = [3,3,1,1,2,2,4,4,5,5]
    cnt1, cnt2, cnt3 = 0,0,0
    
    for i in range(len(answers)):
        s1 = i%5
        s2 = i%8
        s3 = i%10
        
        if n1[s1] == answers[i]:
            cnt1 +=1
        if n2[s2] == answers[i]:
            cnt2 +=1
        if n3[s3] == answers[i]:
            cnt3 +=1
            
    maxNum = max(cnt1,cnt2,cnt3)
    if maxNum == cnt1:
        answer.append(1)
    if maxNum == cnt2:
        answer.append(2)
    if maxNum == cnt3:
        answer.append(3)
        
    return answer

각 방식에서의 순환마디를 고려하여 나머지로 인덱스를 찾아 비교해주었다.

1,2,3번의 수포자가 맞힌 문제 개수를 각 변수에 담아 최대값을 구한 후 결과값을 리턴한다.

▶ 참고사항

enumerate()를 사용해서 해결한 풀이도 있었다.

ex) for i in enumerate(["1", "2"]):

인덱스와 원소를 튜플형식으로 반환한다.

만약, 인덱스와 원소를 각각 구하려면 인자 풀기(unpacking)을 한다.

ex) for i, n enumerate(["1", "2"]):