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

묘걍 2023. 8. 31. 20:15

문제

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

내가 생각한 풀이 방법

supo_1 = [1, 2, 3, 4, 5]의 반복

supo_2 = [2, 1, 2, 3, 2, 4, 2, 5] 의 반복

supo_3 = [3, 3, 1, 1, 2, 2, 4, 4, 5, 5] 의 반복

len(answers)만큼 for 문을 돈다i번 문제에 대해서 정답은 answers[i], 1번 수포자가 찍은 답은 supo_1[i%5], 2번 수포자가 찍은 답은 supo_2 [i%8], 3번 수포자가 찍은 답은 supo_3 [i%10]이다.각각 찍은 답과 정답을 비교하여 맞았다면 사람에 해당하는 변수에 += 1 해준다.

max()로 최대 값 판별 후 answer에 append

코드로 구현

1차 시도

def solution(answers):
    answer = []
    supo_1 = [1, 2, 3, 4, 5]
    supo_2 = [2, 1, 2, 3, 2, 4, 2, 5]
    supo_3 = [3, 3, 1, 1, 2, 2, 4, 4, 5, 5]
    s1, s2, s3 = 0
    for i in range(len(answers)):
        if answers[i] == supo_1[i%5]:
            s1 += 1
        elif answers[i] == supo_2 [i%8]:
            s2 += 1
        elif answers[i] == supo_3 [i%10]:
            s3 += 1
            
    top_score = max(s1, s2, s3)
    answer.append(top_score)
    return answer

- 이 오류는 s1, s2, s3 = 0에서 발생. 이 문장은 0이라는 정수를 s1, s2, s3 세 변수에 동시에 할당하려고 시도. 그러나 정수는 반복 가능한 객체가 아니므로 오류가 발생.

예전에 뭔가 이런거 되는 문장을 봤던 것 같은데ㅠㅠㅠ 뭐였지ㅠㅠ

- 지금 코드는 점수를 반환하고 있음

2차 시도

def solution(answers):
    answer = []
    supo_1 = [1, 2, 3, 4, 5]
    supo_2 = [2, 1, 2, 3, 2, 4, 2, 5]
    supo_3 = [3, 3, 1, 1, 2, 2, 4, 4, 5, 5]
    s1 = 0
    s2 = 0
    s3 = 0
    for i in range(len(answers)):
        if answers[i] == supo_1[i%5]:
            s1 += 1
        elif answers[i] == supo_2 [i%8]:
            s2 += 1
        elif answers[i] == supo_3 [i%10]:
            s3 += 1
            
    top_score = max(s1, s2, s3)
    if top_score == s1:
        answer.append(1)
    if top_score == s2:
        answer.append(2)
    if top_score == s3:
        answer.append(3)
    return answer

- 하나는 맞고 하나는 틀림

- elif 사용 주의.

→elif 조건을 사용하여 수포자의 답변을 체크하면 한 문제에 대해 여러 수포자가 답을 맞추었을 경우 모든 수포자의 점수를 올릴 수 없습니다. 따라서 elif 대신 if를 사용하여 각 수포자의 답변을 독립적으로 체크해야 합니다.

4차 시도

def solution(answers):
    answer = []
    supo_1 = [1, 2, 3, 4, 5]
    supo_2 = [2, 1, 2, 3, 2, 4, 2, 5]
    supo_3 = [3, 3, 1, 1, 2, 2, 4, 4, 5, 5]
    s1 = 0
    s2 = 0
    s3 = 0
    for i in range(len(answers)):
        if answers[i] == supo_1[i%5]:
            s1 += 1
        if answers[i] == supo_2[i%8]:
            s2 += 1
        if answers[i] == supo_3[i%10]:
            s3 += 1
            
    top_score = max(s1, s2, s3)
    if top_score == s1:
        answer.append(1)
    if top_score == s2:
        answer.append(2)
    if top_score == s3:
        answer.append(3)
    return answer