[프로그래머스] 로또의 최고 순위와 최저 순위

 

문제 설명

로또 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라면, 당첨 가능한 최고 순위와 최저 순위의 한 예는 아래와 같습니다.

당첨 번호	31	10	45	1	6	19	결과
최고 순위 번호	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 함수를 완성해주세요.

---

제한사항

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

---

문제 풀이

def solution(lottos, win_nums):
	# 세트로 형변환 하여 중복 제거
    set1 = set(lottos)
    set2 = set(win_nums)
    
    # 0(알아볼 수 없는 번호)
    cnt_zero = lottos.count(0)
    
    # 당첨 가능한 최고 순위
    intersection = len(set(set1 & set2)) + cnt_zero
    # 당첨 가능한 최저 순위
    difference = len(set2.intersection(set1)) 
    
    rank_dict = {6:1, 5:2, 4:3, 3:4, 2:5}
    
    answer1 = rank_dict[intersection] if intersection in rank_dict else 6
    answer2 = rank_dict[difference] if difference in rank_dict else 6
    
    return [answer1, answer2]

 

순서와 상관없이라는 말을 보자마자 일단 set을 사용하는 것을 염두에 두었다.

두개의 수의 집합에서,  최고 순위를 위해 서로 겹치는 교집합을 계산하고,

최저 순위를 위해서는 로또 당첨 번호 집합에서 내가 뽑은 번호 집합을 빼주는 차집합을 구하면 된다는 생각이 들었다.

 

그리고 이 집합을 출력하는게 아니고, 번호를 맞춘 갯수별로 순위가 정해져있기 때문에 이를 위한 딕셔너리를 만들었다.

딕셔너리의 key는 맞춘 갯수, value값을 순위로 설정하였다.

그리고 범위 내의 값이 나온다면 해당 딕셔너리에서 [맞춘 갯수]로 순위(value)를 추출하는 방법을 사용했다.

 

시간 복잡도

• (set(lottos)와 set(win_nums))에서 집합을 생성한다.
  집합으로 변환하는 작업의 시간 복잡도는 O(n) 이므로 이 코드의 시간복잡도는 O(n)이 된다.
• 리스트 lottos의 길이가 n이라고 할 때, count() 메소드는 리스트를 한 번 순회하면서 0의 개수를 세기 때문에 시간 복잡도는 O(n)입니다.

따라서 이 코드의 시간 복잡도는 O(n)이지만, 여기서는 n이 상수이므로 실질적으로는 O(1)입니다.

 

---

다른 풀이

def solution(lottos, win_nums):

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

    cnt_0 = lottos.count(0)
    ans = 0
    for x in win_nums:
        if x in lottos:
            ans += 1
    return rank[cnt_0 + ans],rank[ans]

순위-맞춘 갯수를 맵핑할때,

어차피 우리가 원하는 value값이 숫자이기 때문에 딕셔너리가 아니라 리스트를 사용하는 방법이다!!

밸류값이 숫자인 경우 딕셔너리가 아니라 리스트를 쓰는 방법은 https://kimwoolina.tistory.com/46 에서도 배웠었는데,

막상 문제를 보니 또 생각이 안나고 직관적으로 딕셔너리로 접근하였다.

 

뿐만 아니라, 나는 if else 6 로 낙첨인 경우 값을 지정하는데

낙첨은 어차피 맞춘 갯수가 0이므로,

0인덱스에 6을 넣는 식으로 해결하였다.. 기발하다..!!