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

문제 바로가기

https://school.programmers.co.kr/tryouts/85909/challenges?language=python3

 

💡 문제 요약

수포자 삼인방은 모의고사에 수학 문제를 전부 찍으려 합니다. 수포자는 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 함수를 작성해주세요.

ex)

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

 

💡 알고리즘 설계

• 정답을 맞춘 갯수를 저장할 딕셔너리 right와 세가지 타입의 찍은 답이 저장되어 있는 딕셔너리 ans_dict을 만들어준다.
• 실제 정답지인 answers를 for문을 돌면서 찍은 답과 같을 경우, right의 각 요소에 +1을 해준다.
  • 이 때, 주의할 점은 찍은 답이 계속 반복되도록 해주어야 한다는 점이다.
    • 1번 수포자의 찍은 답은 길이가 5이기 때문에 인덱스에 i%5를 해준다.
    • 2번 수포자의 찍은 답은 길이가 8이기 때문에 인덱스에 i%8을 해준다.
    • 3번 수포자의 찍은 답은 길이가 10이기 때문에 인덱스에 i%10을 해준다.
• 맞춘 갯수가 저장되어 있는 딕셔너리 right에서 가장 많이 맞춘 갯수와 1,2,3번 수포자의 각 value 값이 같은 경우, 해당 value의 key값을 res 리스트에 저장해준다.

💡 내 코드

def solution(answers):
    right = {1:0, 2:0, 3:0}    # 정답을 맞춘 갯수
    ans_dict={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]}    
    
    for i in range(len(answers)):
        if ans_dict[1][i%5] == answers[i]:
            right[1]+=1
        if ans_dict[2][i%8] == answers[i]:
            right[2]+=1
        if ans_dict[3][i%10] == answers[i]:
            right[3]+=1
        
    res=[]
    for k, v in right.items():
        if max(right.values()) == v:
            res.append(k)
    
    
    return res

 

💡 오답노트

• 찍은 답이 계속 반복되도록 해주어야 하는 것을 간과하고 실행 했더니 런타임 에러가 났다.