Python 을 이용한 프로그래머스 LV 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문제씩을 맞췄습니다.

 

import math
def solution(answers):
    answer = []
    list1 = '12345'
    list2 = '21232425'
    list3 = '3311224455'
    cnt1 = 0
    cnt2 = 0
    cnt3 = 0
    result = []
    if len(list1) < len(answers):
        while len(list1) < len(answers):
            list1 = list1 * math.ceil((len(answers) / 5))
    if len(list2) < len(answers):
        while len(list2) < len(answers):
            list2 = list2 * math.ceil((len(answers) / 8))

    if len(list3) < len(answers):
        while len(list3) < len(answers):
            list3 = list3 * math.ceil((len(answers) / 10))

    for i in range(0,len(answers)):
        if answers[i]==int(list1[i:i+1]): cnt1 += 1
        if answers[i]==int(list2[i:i+1]): cnt2 += 1
        if answers[i]==int(list3[i:i+1]): cnt3 += 1
        
    if (cnt1>cnt2 and cnt1>cnt3):
        answer.append(1)
    elif (cnt2>cnt1 and cnt2>cnt3):
        answer.append(2)
    elif (cnt3>cnt1 and cnt3>cnt2 ):
        answer.append(3)
    elif (cnt1==cnt2==cnt3):
        answer.append(1)
        answer.append(2)
        answer.append(3)
    elif (cnt2==cnt1>cnt3):
        answer.append(1)
        answer.append(2)
    elif (cnt2==cnt3>cnt1):
        answer.append(2)
        answer.append(3)
    elif (cnt1==cnt3>cnt2):
        answer.append(1)
        answer.append(3)
    return answer
solution([1,2,3,4,5,2,1,4,5,2,2,2,5,1,5,5,2,1,4])

음.. 내가 봐도 조금 난해한 코드다.

 

이걸 어떻게 풀었을까.

 

우선 수포자들의 정답 패턴을 list로 표현할지, 아니면 문자열로 표현해서 형변환 시켜줄지 생각을 해봤는데,

나는 아직 파이썬의 문법을 잘 모르는 거 같다.

 

그리고 아무래도 답안 패턴 길이를 문자랑 좀 맞추고 싶었는데, 굳이 안해줘도 답 나오는 데는 이상이 없으니 그냥 뒀다.

 

답안 패턴 길이가 문제 답안 길이보다 짧다면

문제 길이/답안 패턴 길이를 한 값을 답안 패턴 길이랑 곱해줘라. 라고 하면 최소 문제길이보단 길고, 최소한의 패턴 반복이 이루어진다.

 if len(list1) < len(answers):
        while len(list1) < len(answers):
            list1 = list1 * math.ceil((len(answers) / 5))
    if len(list2) < len(answers):
        while len(list2) < len(answers):
            list2 = list2 * math.ceil((len(answers) / 8))

    if len(list3) < len(answers):
        while len(list3) < len(answers):
            list3 = list3 * math.ceil((len(answers) / 10))

맞다. 그리고 나눠진 값이 float형태면 곱해지지 않아서 올림함수를 이용해서 올려줬다.

 

 

list로 표현했을 때 에러가 좀 나더라. 그래서 그냥 문자열로 표현했다.

또한 정답을 비교해야겠다고 생각했는데, 이부분도 list가 좀 더 적합하다고 생각했다.

for문을 돌려서 각 자리를 비교하면 되겠지 했는데, 또 잘 안됐다. 

 

그리고 이제 답안을 비교할 차례이다.

for i in range(0,len(answers)):
        if answers[i]==int(list1[i:i+1]): cnt1 += 1
        if answers[i]==int(list2[i:i+1]): cnt2 += 1
        if answers[i]==int(list3[i:i+1]): cnt3 += 1

이렇게 했다.

answers 리스트에 담긴 값들은 숫자형이니 비교할 친구들을 int로 형변환 시켰다. 그리고 대망의 쓸데없는 코드..

 

if (cnt1>cnt2 and cnt1>cnt3):
        answer.append(1)
    elif (cnt2>cnt1 and cnt2>cnt3):
        answer.append(2)
    elif (cnt3>cnt1 and cnt3>cnt2 ):
        answer.append(3)
    elif (cnt1==cnt2==cnt3):
        answer.append(1)
        answer.append(2)
        answer.append(3)
    elif (cnt2==cnt1>cnt3):
        answer.append(1)
        answer.append(2)
    elif (cnt2==cnt3>cnt1):
        answer.append(2)
        answer.append(3)
    elif (cnt1==cnt3>cnt2):
        answer.append(1)
        answer.append(3)
    print(answer)

좀 난감하다.. 이거를 내가 어떻게 구현해야할까 생각하다가 그냥 헤딩한다고 생각하고 그냥 막 썼는데. 되긴 되더라.

 

아무튼 나는 이 코드를 조금 수정하려고 한다. 너무 난잡하지 않는가?

 

보니깐 enumerate라는 함수를 쓰더라. 이게 뭘까?

 

enumerate함수는 반복문 사용 시 몇 번째 반복문인지 확인이 필요할 수 있는데, 이때 사용한다고 한다.

인덱스 번호와 컬렉션 원소를 tuple 형태로 반환한다고 한다. 음..

import math
def solution(answers):
    answer = []
    list1 = [1,2,3,4,5]
    list2 = [2,1,2,3,2,4,2,5]
    list3 = [3,3,1,1,2,2,4,4,5,5]
    score = [0, 0, 0]

    for idx, ans in enumerate(answers):
        if ans==list1[idx%len(list1)]: score[0] += 1
        if ans==list2[idx%len(list2)]: score[1] += 1
        if ans==list3[idx%len(list3)]: score[2] += 1

    for idx, s in enumerate(score):
        if s == max(score):
            answer.append(idx+1)

    return answer

솔직히 조금 민망하다. 나는 뭘 그렇게 열심히 적어댔을까. 적기 전에 생각하는 버릇을 들여야겠다...

 

idx, ans를 answers 에서 가져온다.

그리고 길이가 넘어가는 것을 생각해 몫을 가져와 list들의 사이클을 돌린다.

 

그리고 마지막으로 내가 노가다 했던 것을 저렇게 단 세 줄의 코드로 완성시켰는데,

 

score 점수를 가져와서 max를 돌린 값과 같다면, answer에 해당 번수를 넣어준다. idx는 0부터 시작하니 오름차순으로 따로 정렬할 필요가 없다.

 

 

 

출처 https://programmers.co.kr/learn/courses/30/lessons/42840

코딩테스트 연습 - 모의고사