<프로그래머스 문제풀이: 모의고사> Level 1 - 파이썬

 

가장 먼저 이 문제를 보았을때는 단순 노가다 문제일 것이라고 직감했다. 따라서 그냥 단순하게 모든 경우의 수를 다 때려박았다:

def solution(answers):
    
    A = [1, 2, 3, 4, 5] * 2000
    B = [2, 1, 2, 3, 2, 4, 2, 5] * 1250
    C = [3, 3, 1, 1, 2, 2, 4, 4, 5, 5] * 1000
    
    numA = 0
    numB = 0
    numC = 0
    
    for i in range(len(answers)):
        if answers[i] == A[i]:
            numA += 1
        else:
            pass
        if answers[i] == B[i]:
            numB += 1
        else:
            pass
        if answers[i] == C[i]:
            numC += 1
        else:
            pass
    
    numlist = [numA, numB, numC]
    
    if numA != numB != numC:
        if max(numlist) == numA:
            answer = [1]
        if max(numlist) == numB:
            answer = [2]
        if max(numlist) == numC:
            answer = [3]
    elif numA == numB == numC:
        answer = [1, 2, 3]
    elif numA == numB and numA != numC:
        if max(numlist) == numA:
            answer = [1, 2]
        elif max(numlist) == numC:
            answer = [3]
    elif numC == numB and numA != numC:
        if max(numlist) == numA:
            answer = [1]
        elif max(numlist) == numC:
            answer = [2, 3]
    elif numA == numC and numA != numB:
        if max(numlist) == numA:
            answer = [1, 3]
        elif max(numlist) == numB: 
            answer = [2]
    return answer

 

보다 싶이 매우 길고 복잡하다. 실제로 제출을 해보니 걸리는 시간도 꽤 긴 편이었다. 정확성 테스트에서는 통과할 줄 알았는데 테스트2와 테스트12를 통과하지 못했다. 아무리 찾아도 이유가 잘 안보였다. 따라서 그냥 약간 다른 방식으로 풀기로 했다:

def solution(answers):
    
    A = [1, 2, 3, 4, 5] * 2000
    B = [2, 1, 2, 3, 2, 4, 2, 5] * 1250
    C = [3, 3, 1, 1, 2, 2, 4, 4, 5, 5] * 1000
    
    numA = 0
    numB = 0
    numC = 0
    
    for i in range(len(answers)):
        if answers[i] == A[i]:
            numA += 1
        else:
            pass
        if answers[i] == B[i]:
            numB += 1
        else:
            pass
        if answers[i] == C[i]:
            numC += 1
        else:
            pass
    
    numlist = [numA, numB, numC]
    anwer = []
    
    for i in range(len(numlist)):
        if numlist[i] == max(numlist):
            answer.append(i+1)
            
    return answer

 

일일이 다 체크 할 필요 없이, max(numlist)를 이용해서 numlist[i]번째 수와 같은지 확인하고, 같을 경우엔 answer list에 추가시켰다.

 

 

다른 사람의 풀이:

def solution(answers):
    pattern1 = [1,2,3,4,5]
    pattern2 = [2,1,2,3,2,4,2,5]
    pattern3 = [3,3,1,1,2,2,4,4,5,5]
    score = [0, 0, 0]
    result = []

    for idx, answer in enumerate(answers):
        if answer == pattern1[idx%len(pattern1)]:
            score[0] += 1
        if answer == pattern2[idx%len(pattern2)]:
            score[1] += 1
        if answer == pattern3[idx%len(pattern3)]:
            score[2] += 1

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

    return result

프로그래머스에서 좋아요를 가장 많이 받은 풀이이다. 여기서 이 개발자는 enumerate과 나머지 값을 반환 시켜주는 %를 사용하여 풀었다. 문제의 설명에 최대 10,000의 문제가 있을 수 있다고 말한 것을 이용해 모든 숫자를 list에 넣어주고 시작한게 아니라, 나머지 방식을 이용하여 비교적 더 간단하게 풀었다.