[완전탐색] 모의고사 - level1

[heerucan]

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

def solution(answers):
    total = {1: 0, 2: 0, 3: 0}

    one = [1,2,3,4,5]*2000
    two = [2,1,2,3,2,4,2,5]*1300
    three = [3,3,1,1,2,2,4,4,5,5]*1000
    
    for i in range(len(answers)):
        answer = answers[i]
        if one[i] == answer:
            total[1] += 1
        if two[i] == answer:
            total[2] += 1
        if three[i] == answer:
            total[3] += 1
    
    # 딕셔너리를 정렬
    answer = []
    sortedTotal = dict(sorted(total.items()))

    k = max([sortedTotal[1], sortedTotal[2], sortedTotal[3]])

    if k == sortedTotal[1]:
        answer.append(1)
    if k == sortedTotal[2]:
        answer.append(2)
    if k == sortedTotal[3]:
        answer.append(3)

    return answer

solution([1,2,3,4,5,1,2,3,4,5])

새로 알게 된 점

• 딕셔너리 value 기준으로 오름차순 정렬하면 튜플로 반환되는데 그걸 다시 딕셔너리화하면
• • dict(sorted(total.items()))
• key 기준 내림차순 정렬하면 {'b': 0, 'a': 1, 'c': 2}에서 ['a', 'b', 'c']로 반환
• • sorted(total, reverse=True)
• 배열 자르기 -> num[0:2] 이렇게 하면 [1,2] 앞에서 2개만 들어옴