[프로그래머스 level1] 로또의 최고 순위와 최저 순위 - 파이썬

💡 문제 설명

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

순위 당첨 내용

1	6개 번호가 모두 일치
2	5개 번호가 일치
3	4개 번호가 일치
4	3개 번호가 일치
5	2개 번호가 일치
6(낙첨)	그 외

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

당첨 번호3110451619결과

최고 순위 번호	31	0→10	44	1	0→6	25	4개 번호 일치, 3등
최저 순위 번호	31	0→11	44	1	0→7	25	2개 번호 일치, 5등

• 순서와 상관없이, 구매한 로또에 당첨 번호와 일치하는 번호가 있으면 맞힌 걸로 인정됩니다.
• 알아볼 수 없는 두 개의 번호를 각각 10, 6이라고 가정하면 3등에 당첨될 수 있습니다.
  • 3등을 만드는 다른 방법들도 존재합니다. 하지만, 2등 이상으로 만드는 것은 불가능합니다.
• 알아볼 수 없는 두 개의 번호를 각각 11, 7이라고 가정하면 5등에 당첨될 수 있습니다.
  • 5등을 만드는 다른 방법들도 존재합니다. 하지만, 6등(낙첨)으로 만드는 것은 불가능합니다.

민우가 구매한 로또 번호를 담은 배열 lottos, 당첨 번호를 담은 배열 win_nums가 매개변수로 주어집니다. 이때, 당첨 가능한 최고 순위와 최저 순위를 차례대로 배열에 담아서 return 하도록 solution 함수를 완성해주세요.

 

💡 나의 풀이

def solution(lottos, win_nums):
    counter = 0
    answer = []
    for lotto in lottos:
        if win_nums.count(lotto) > 0:
            counter += 1
    max = counter + lottos.count(0)
    min = counter

    if max == 6:
        answer.append(1)
    elif max == 5:
        answer.append(2)
    elif max == 4:
        answer.append(3)
    elif max == 3:
        answer.append(4)
    elif max == 2:
        answer.append(5)
    else:
        answer.append(6)

    if min == 6:
        answer.append(1)
    elif min == 5:
        answer.append(2)
    elif min == 4:
        answer.append(3)
    elif min == 3:
        answer.append(4)
    elif min == 2:
        answer.append(5)
    else:
        answer.append(6)

    return answer

처음 푼 나의 풀이를 보면 정말 더럽다고 볼 수 있다..  부끄러울 정도

 

먼저, win_nums와 lottos의 같은 수를 찾고, lottos의 0의 개수에 따라 순위가 바뀌기 때문에

최고 순위는 같은 수 + lottos.count(0)으로 하고, 최저 순위는 같은 수로 지정을 했다.

 

 그러고 나서 맞은 개수에 따라 로또 순위를 출력해야 하는데 이 부분이 고민이 되었다

그냥 무작정 if문으로 지저분하게 만들었더니 통과는 했지만, 이렇게 풀이하면 안 되기에 다른 사람의 문제를 살펴본 결과

다들 딕셔너리를 이용해서 깔끔하게 해결을 했다..

 

어제 딕셔너리 잊지 말자 했는데 하루 만에 잊었다.. 반성하자

 

다시 딕셔너리를 이용하여 코드를 수정했다.

💡 풀이 수정 (딕셔너리 이용)

def solution(lottos, win_nums):
    counter = 0
    answer = []
    dic = {6: 1, 5: 2, 4: 3, 3: 4, 2: 5, 1: 6, 0: 6}
    for lotto in lottos:
        if win_nums.count(lotto) > 0:
            counter += 1
    answer.append(dic[counter + lottos.count(0)])
    answer.append(dic[counter])
    return answer

좀 더 깔끔해졌다

 

하지만 다른 사람 풀이를 보니 더 좋은 방법도 있었다

바로 리스트 인덱스 번호를 이용하면 된다.

💡 제일 간단한 방법

rank=[6,6,5,4,3,2,1]

 

이렇게 하면 정답 개수를 인덱스로 해서 출력하면 끝.

왜 생각이 안 났을까..