N개의 최소공배수 [Python]

URL https://programmers.co.kr/learn/courses/30/lessons/77484

문제 설명

• 로또 6/45(이하 ‘로또’로 표기)는 1부터 45까지의 숫자 중 6개를 찍어서 맞히는 대표적인 복권입니다. 아래는 로또의 순위를 정하는 방식입니다. 1

• 로또를 구매한 민우는 당첨 번호 발표일을 학수고대하고 있었습니다. 하지만, 민우의 동생이 로또에 낙서를 하여, 일부 번호를 알아볼 수 없게 되었습니다. 당첨 번호 발표 후, 민우는 자신이 구매했던 로또로 당첨이 가능했던 최고 순위와 최저 순위를 알아보고 싶어 졌습니다.
• 알아볼 수 없는 번호를 0으로 표기하기로 하고, 민우가 구매한 로또 번호 6개가 44, 1, 0, 0, 31 25라고 가정해보겠습니다. 당첨 번호 6개가 31, 10, 45, 1, 6, 19라면, 당첨 가능한 최고 순위와 최저 순위의 한 예는 아래와 같습니다.

• 순서와 상관없이, 구매한 로또에 당첨 번호와 일치하는 번호가 있으면 맞힌 걸로 인정됩니다.
• 알아볼 수 없는 두 개의 번호를 각각 10, 6이라고 가정하면 3등에 당첨될 수 있습니다.
• 3등을 만드는 다른 방법들도 존재합니다. 하지만, 2등 이상으로 만드는 것은 불가능합니다.
• 알아볼 수 없는 두 개의 번호를 각각 11, 7이라고 가정하면 5등에 당첨될 수 있습니다.
• 5등을 만드는 다른 방법들도 존재합니다. 하지만, 6등(낙첨)으로 만드는 것은 불가능합니다.
• 민우가 구매한 로또 번호를 담은 배열 lottos, 당첨 번호를 담은 배열 win_nums가 매개변수로 주어집니다. 이때, 당첨 가능한 최고 순위와 최저 순위를 차례대로 배열에 담아서 return 하도록 solution 함수를 완성해주세요.

제한사항

• lottos는 길이 6인 정수 배열입니다.
• lottos의 모든 원소는 0 이상 45 이하인 정수입니다.
• 0은 알아볼 수 없는 숫자를 의미합니다.
• 0을 제외한 다른 숫자들은 lottos에 2개 이상 담겨있지 않습니다.
• lottos의 원소들은 정렬되어 있지 않을 수도 있습니다.
• win_nums은 길이 6인 정수 배열입니다.
• win_nums의 모든 원소는 1 이상 45 이하인 정수입니다.
• win_nums에는 같은 숫자가 2개 이상 담겨있지 않습니다.
• win_nums의 원소들은 정렬되어 있지 않을 수도 있습니다.

문제풀이

• result에 등수를 미리 저장한다. (6등은 1개 혹은 0개 맞춘것이기 때문에 일단 제외)
• low, high에 맞춘 갯수를 저장하기 위해 0으로 셋팅 해준다.
• 최종 답변이 될 answer리스트를 만든다.
• high 등수에서 0에서 바꿔줄 못맞춘 not_matched의 리스트를 만들어 준다.
• 경우의 수
  • 1 : 0이 모두 로또번호와 맞을때 (0을 내가 선택하지 않고, 로또당첨번호로 치환해주어야함)
  • 2 : 0이 모두 로또번호와 맞지 않을떄 (0을 굳이 다른수로 치환할 필요는 없음)
  • 3 : 0이 없고, 로또 번호가 모두 맞지 않을때

• 위의 경우의 수를 고려하여 코드를 작성하면 된다.

• 최저 등수
  • 일단 최저 등수는 0으로 된것도 하나도 안맞추면 되기 때문에 다른 전처리 없이 바로 for문으로 확인한다.
  • low_rank에 맞춘 숫자의 갯수를 저장하고, 이 갯수에 따라 등수를 result에서 가져온다.
  • 만일 low_rank가 0,1이면 answer에 6을 append하는 if문을 작성 뒤, 해당 사항이 없으면 실제 등수를 result에서 가져온다.
• 최고 등수
  • 최고 등수는 0을 모두 맞춘것으로 가정하면 된다.
  • 일단 내가 맞추지 못한 1등 번호를 not_matched의 리스트에 넣어준다
  • 이후 0을 not_matched의 맨 앞의 번호로 변경시켜주고, not_matched에는 맨 앞의 번호를 제거한다 (Pop사용)
  • 위의 과정을 0이 없어질때까지 계속 한다.
  • 0을 모두 선택된 번호로 치환했다면 최저 등수를 구한것처럼 for문을 돌며 맞춘 번호의 갯수를 high_rank에 더한다.
  • 만일 high_rank에 0,1이면 answer에 6을 append하는 if문을 작성 뒤, 해당 사항이 없으면 실제 등수를 result에서 가져온다.

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38

def solution(lottos, win_nums):
    result = {6:1, 5:2, 4:3, 3:4, 2:5}
    low_rank = 0
    high_rank = 0
    answer = []
    not_matched = []
    lottos = sorted(lottos)

    # get low rank
    for i in lottos:
        if i in win_nums:
            low_rank += 1
    
    if low_rank < 2:
        answer.append(6)
    else:
        answer.append(result[low_rank])

    # get high_rank
    for i in win_nums:
        if i not in lottos:
            not_matched.append(i)

    for idx, lotto in enumerate(lottos):
        if lotto == 0:
            lottos[idx] = not_matched[0]
            not_matched.pop(0)

    for i in lottos:
        if i in win_nums:
            high_rank += 1
            
    if high_rank < 2:
        answer.append(6)
    else:
        answer.append(result[high_rank])        

    return sorted(answer)

1
2
3

lottos = [44, 1, 0, 0, 31, 25]
win_nums = [31, 10, 45, 1, 6, 19]
solution(lottos, win_nums)

1

[3, 5]

1
2
3

lottos = [0, 0, 0, 0, 0, 0]
win_nums = [38, 19, 20, 40, 15, 25]
solution(lottos, win_nums)

1

[1, 6]

1
2
3

lottos = [45, 4, 35, 20, 3, 9]
win_nums = [20, 9, 3, 45, 4, 35]
solution(lottos, win_nums)

1

[1, 1]