🔗 문제 링크

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

💻 코드

def solution(answers):
    answer = []
    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] * 3
    
    for i in range(1, len(answers) + 1):
        if answers[i - 1] == pattern1[(i - 1) % len(pattern1)]:
            score[0] += 1
        if answers[i - 1] == pattern2[(i - 1) % len(pattern2)]:
            score[1] += 1
        if answers[i - 1] == pattern3[(i - 1) % len(pattern3)]:
            score[2] += 1
    
    max_result = max(score)

    for i in range(0, 3):
        if max_result == score[i]:
            answer.append(i + 1)
    # for idx, i in enumerate(score):
    #     if i == max(score):
    #         answer.append(idx + 1)
    
    return answer

📝 정리

파이썬 enumerate()

• 인덱스 번호와 컬렉션의 원소를 tuple 형태로 반환

t = [1, 5, 7, 33, 39, 52]
for p in enumerate(t):
     print(p)
 
(0, 1)
(1, 5)
(2, 7)
(3, 33)
(4, 39)
(5, 52)