[Programmers][Python] 로또의 최고 순위와 최저순위(Lv.1)
1. 문제
문제 설명
로또 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등(낙첨)으로 만드는 것은 불가능합니다.
3등을 만드는 다른 방법들도 존재합니다. 하지만, 2등 이상으로 만드는 것은 불가능합니다. 알아볼 수 없는 두 개의 번호를 각각 11, 7이라고 가정하면 5등에 당첨될 수 있습니다. 5등을 만드는 다른 방법들도 존재합니다. 하지만, 6등(낙첨)으로 만드는 것은 불가능합니다.
제한사항
- lottos는 길이 6인 정수 배열입니다.
- lottos의 모든 원소는 0 이상 45 이하인 정수입니다.
- 0은 알아볼 수 없는 숫자를 의미합니다.
- 0을 제외한 다른 숫자들은 lottos에 2개 이상 담겨있지 않습니다.
- lottos의 원소들은 정렬되어 있지 않을 수도 있습니다.
- win_nums은 길이 6인 정수 배열입니다.
- win_nums의 모든 원소는 1 이상 45 이하인 정수입니다.
- win_nums에는 같은 숫자가 2개 이상 담겨있지 않습니다.
- win_nums의 원소들은 정렬되어 있지 않을 수도 있습니다.
입출력 예
| lottos | win_nums | result |
|---|---|---|
| [44, 1, 0, 0, 31, 25] | [31, 10, 45, 1, 6, 19] | [3, 5] |
| [0, 0, 0, 0, 0, 0] | [38, 19, 20, 40, 15, 25] | [1, 6] |
| [45, 4, 35, 20, 3, 9] | [20, 9, 3, 45, 4, 35] | [1, 1] |
입출력 예 #1
문제 예시와 같습니다.
입출력 예 #2
알아볼 수 없는 번호들이 아래와 같았다면, 1등과 6등에 당첨될 수 있습니다.
| 당첨 번호 | 38 | 19 | 20 | 40 | 15 | 25 | 결과 |
|---|---|---|---|---|---|---|---|
| 최고 순위 번호 | 0→38 | 0→19 | 0→20 | 0→40 | 0→15 | 0→25 | 6개 번호 일치, 1등 |
| 최저 순위 번호 | 0→21 | 0→22 | 0→23 | 0→24 | 0→26 | 0→27 | 0개 번호 일치, 6등 |
입출력 예 #3
민우가 구매한 로또의 번호와 당첨 번호가 모두 일치하므로, 최고 순위와 최저 순위는 모두 1등입니다.
2. 풀이
최고 순위와 최저 순위를 구하기 위해 일치하는 번호 개수의 min, max를 먼저 구한다. 0으로 표시한 숫자가 모두 틀린다는 가정하에 로또 번호 중 당첨 번호와 일치하는 개수가 min이 되고, 최고 순위는 0으로 표시한 숫자가 모두 맞는다는 가정이기 때문에 여기에 0의 개수를 더해주면 max가 된다. 따라서 구해야할 값은 다음과 같다.
- 일치하는 숫자 개수
- 0의 개수
우선 0의 개수는 리스트 내장 함수 count를 사용하여 간단하게 구할 수 있다. list.count(x)의 형태로 사용하며 해당 리스트에 포함된 x의 개수를 구하여 리턴한다.
일치하는 숫자 개수는 단순하게 생각하면 if문을 사용하여 하나하나 비교할 수 있지만, 더 간단한 방법은 set의 교집합으로 구하는 것입니다. 주어진 리스트를 set으로 변환한 후 intersection를 사용하여 교집합을 구하고 len()함수로 개수를 구한다.
마지막으로, 일치하는 번호 개수를 순위로 바꾸는 방법으로는 딕셔너리를 사용했다. key,value가 개수,순위로 구성된 딕셔너리를 만들어 인덱싱한다.
코드1 ❌
def solution(lottos, win_nums):
zeros = lottos.count(0)
cnt = len(set(lottos).intersection(win_nums))
answer = [7 - (zeros + cnt), 7 - cnt]
if answer[1] > 6:
answer[1] = 6
return answer
처음에는 순위를 7 - 일치 개수로 구하고, 0~1개 일치가 모두 6등이므로 그 부분을 if문으로 처리해주었는데 정확성 테스트에서 1개를 통과하지 못했다.
코드2
def solution(lottos, win_nums):
rank_dict = {0:6, 1:6, 2:5, 3:4, 4:3, 5:2, 6:1}
zeros = lottos.count(0)
cnt = len(set(lottos).intersection(win_nums))
answer = [rank_dict[zeros + cnt], rank_dict[cnt]]
return answer