[프로그래머스] 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번 수포자의 경우, 1번부터 순서대로 찍기 때문에 제일 어렵지도 않고
제일 무난하게 알고리즘을 짤 수 있다.

하지만 문제는 2번 수포자 부터였다.
연속으로 2번을 찍는 경우는 없고, 2번을 먼저 찍고 난 다음의 숫자는 1부터 1씩 커지는 경우이다.
이를 알고리즘으로 구현하기 위해서,

무조건 홀수 번호 문제에서는 2를 찍게되고
짝수 번호의 문제에서는 1부터 찍게 되는데
이를 위해 for문의 앞단에 다음과 같은 조건을 추가했다.

for i in range(0, len(answers)):
        if correct2 == 2:
            correct2 = 3

홀수 번호의 문제가 2가되는 경우 3으로 강제적으로 바꾸어 주었다.

다음은 가장 난해한 3번 수포자이다.
여기서는 같은 숫자를 연속으로 두 번 찍고 1씩 증가시키는데,
이 때 2다음은 3이 아니라 4이다.
그래서 이를 해결하기 위해 같은 숫자가 2번 나왔는지 검사하는 스위치 변수를 추가했다.
if문을 여러개 써서 다음 숫자가 무엇인지 파악하고 강제적으로 바꾸는 코드인데,
이 부분이 지저분해서 다시 풀어볼 예정이다.

코드

def solution(answers):
    answer = []
    temp = []
    student1 = 0
    student2 = 0
    student3 = 0

    # 수포자 1
    correct = 1
    for i in range(0, len(answers)):
        if correct > 5:
            correct = 1
        if correct == answers[i]:
            student1 += 1
        correct += 1
    temp.append(student1)
    
    # 수포자 2
    correct1 = 2
    correct2 = 1
    for i in range(0, len(answers)):
        if correct2 == 2:
            correct2 = 3
        if correct2 > 5:
            correct2 = 1
        if i % 2 == 0:
            if answers[i] == correct1:
                student2 += 1
        else:
            if answers[i] ==  correct2:
                student2 += 1
            correct2 += 1
    temp.append(student2)

    # 수포자 3
    switch = 0 # 같은 숫자가 두 번 나왔는 지 검사하는 스위치
    correct = 3
    for i in range(0, len(answers)):
        if answers[i] == correct:
            student3 += 1
        if correct == 3 and switch == 1:
            correct = 1
        elif correct == 5 and switch == 1:
            correct = 3
        elif correct == 2 and switch == 1:
            correct = 4
        elif (correct == 1 or correct == 4) and switch == 1:
            correct += 1
        if switch == 0:
            switch = 1
        elif switch == 1:
            switch = 0
    temp.append(student3)
    
    max = student1
    for i in range(1,3):
        if temp[i] >= max:
            max = temp[i]
    
    if max == student1:
        answer.append(1)
    if max == student2:
        answer.append(2)
    if max == student3:
        answer.append(3)

    return answer