모의고사 (완전탐색, enumerate)

문제 설명

수포자는 수학을 포기한 사람의 준말입니다. 수포자 삼인방은 모의고사에 수학 문제를 전부 찍으려 합니다. 수포자는 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문제씩을 맞췄습니다.

---

p1 = [1, 2, 3, 4, 5] * 2000
p2 = [2, 1, 2, 3, 2, 4, 2, 5] * 1250
p3 = [3, 3, 1, 1, 2, 2, 4, 4, 5, 5] * 1000

def solution(answers):
    answer = []
    countP1 = 0
    countP2 = 0
    countP3 = 0
    for i in range(0, len(answers)): 	# 수포자들의 정답수를 확인하기 위한 반복문
        if (p1[i] == answers[i]):
            countP1 += 1
        if (p2[i] == answers[i]):
            countP2 += 1
        if (p3[i] == answers[i]):
            countP3 += 1
            
    _max = max(countP1, countP2, countP3)	#가장 많이 맞춘 수포자 찾기
    if _max == countP1:
        answer = [1]
        if countP1 == countP3:
            answer = [1, 3]
        if countP1 == countP2: 
            answer = [1, 2]
            if countP2 == countP3:
                answer = [1, 2, 3]
    elif _max == countP2:
        answer = [2]
        if countP2 == countP3:
            answer = [2, 3]
    elif _max == countP3:
        answer = [3]
        
    return answer

저는 수포자들의 정답 수를 확인하고 max함수로 최대값을 찾은 후 최대값을 가지고 있는 수포자를 찾는 방식으로 답을 구해봤습니다. 그치만 뭔가 지저분해 보이는 걸 견딜 수 없어 충분히 더 깔끔한 코드를 찾아 보았습니다. 프로그래머스는 잘짠 코드를 볼 수 있어서 좋은것 같습니다. 

---

def solution(answers):
    p1 = [1,2,3,4,5]
    p2 = [2,1,2,3,2,4,2,5]
    p3 = [3,3,1,1,2,2,4,4,5,5]
    count = [0, 0, 0]
    result = []

    for index, answer in enumerate(answers):
        if answer == p1[index%len(p1)]:
            count[0] += 1
        if answer == p2[index%len(p2)]:
            count[1] += 1
        if answer == p3[index%len(p3)]:
            count[2] += 1

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

    return result

enumerate라는걸 사용했네요. enumerate는 반복문 사용시 리스트의 인덱스번호와 원소를 튜플의 형태로 반환합니다.

예를 들면

 a = [10, 20, 30, 40]
 for i in enumerate(a):
 	print(i)
---------
(0, 10)
(1, 20)
(2, 30)
(3, 40)

마치 dict의 key-value 형태와 비슷해 보이네요. 다음으로 눈에 보이는 특징은 p1[index%len(p3)] 저는 시험의 최대 문제는 10000문제로 구성되어 있기때문에 수포자들의 리스트를 모두 10000개로 만들어 주었지만 저렇게 사용하면 10000개를 다 만들어 주지 않아도 순환하면서 읽어올 수 있겠네요.

이번 문제에서 코드를 줄이는 keypoint는 enumerate인것 같습니다.