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

목차

• 1. 문제설명
• 2. 제한사항
• 3. 파이썬 풀이
• 4. 파이썬 풀이 설명
• 5. 시행착오
• 6. 느낀점

---

1. 문제설명

민수는 로또 번호를 구매하고, 당첨이 되었는지 확인하려 합니다. 하지만 동생이 낙서를 해서 일부 번호가 알아볼 수 없게 되었는데, 이를 0으로 표기하였고, 당첨 로또 번호 6개 자리에서 실제 민우가 적은 로또 번호 6개랑 비교를 하여 등수를 알아보도록 합니다. 이때 0으로 표기한 알아볼 수 없는 번호 같은 경우, 최고 순위 번호, 최저 순위 번호로서 2가지 경우로, 실제 민우가 0으로 적은 경우가 모두 로또 번호와 일치하는 경우와, 그렇지 않을 경우, 민우의 등수에 대해서 알아보고 민우의 등수를 최고 순위와 최저 순위를 result 값으로 [3,5] 이런식으로 반환합니다. 최고 순위는 3등, 최저 순위는 5등으로서, 0이 2번 나왔으며 민우가 최대 4개까지 맞췄음을 알 수 있습니다.

 

2. 제한사항

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

3. 파이썬 풀이

def solution(lottos, win_nums):
    # 집합으로 문제 풀이
    cnt = len(set(lottos) & set(win_nums))
    
    # 0 이 있는 경우
    cnt_0 = lottos.count(0)
    
    # 최고 등수와 최저 등수 표현
    # 1등일 경우 모두 맞춰야 함, cnt = 6 일 때 등수는 1임
    # 2등일 경우 5개 맞춰야함, cnt = 5 일 때 등수는 2임
    
    # 0 이 있는 경우 최고 등수는 0의 개수만큼 증가됨
    # 1등일 경우, 6개 모두 맞추는 경우에서, 0이 1개 포함된 경우, 일반 cnt = 5, 0이 있는 경우 = 1, 전체 맞춘 cnt = 6일 때 등수는 1임
    
    # 낙첨일 경우, 6개 중 0개 또는 1개만 맞추는 경우임
    
    # 위의 조건을 종합하면
    max_rank = 7 - max(cnt+cnt_0,1)
    min_rank = 7 - max(cnt, 1)
    
    answer = [max_rank, min_rank]
    return answer

파이썬 풀이 결과

테스트 1
입력값 〉	[44, 1, 0, 0, 31, 25], [31, 10, 45, 1, 6, 19]
기댓값 〉	[3, 5]
실행 결과 〉	테스트를 통과하였습니다.
테스트 2
입력값 〉	[0, 0, 0, 0, 0, 0], [38, 19, 20, 40, 15, 25]
기댓값 〉	[1, 6]
실행 결과 〉	테스트를 통과하였습니다.
테스트 3
입력값 〉	[45, 4, 35, 20, 3, 9], [20, 9, 3, 45, 4, 35]
기댓값 〉	[1, 1]
실행 결과 〉	테스트를 통과하였습니다.

채점 결과

테스트 1 〉	통과 (0.01ms, 10.3MB)
테스트 2 〉	통과 (0.01ms, 10.2MB)
테스트 3 〉	통과 (0.01ms, 10.2MB)
테스트 4 〉	통과 (0.01ms, 10.2MB)
테스트 5 〉	통과 (0.01ms, 10.1MB)
테스트 6 〉	통과 (0.01ms, 10.2MB)
테스트 7 〉	통과 (0.01ms, 10.2MB)
테스트 8 〉	통과 (0.01ms, 10.2MB)
테스트 9 〉	통과 (0.01ms, 10.3MB)
테스트 10 〉	통과 (0.01ms, 10.2MB)
테스트 11 〉	통과 (0.01ms, 10.1MB)
테스트 12 〉	통과 (0.01ms, 10.2MB)
테스트 13 〉	통과 (0.01ms, 10.2MB)
테스트 14 〉	통과 (0.01ms, 10.2MB)
테스트 15 〉	통과 (0.00ms, 10.1MB)

4. 파이썬 풀이 설명

위 문제의 주요 특징은 2가지 경우가 발생한다는 점입니다

 

0이 나오는 경우와 나오지 않는 경우입니다. 이에따라 등수가 달라질 수 있습니다.

그래서 우선 0이 나오지 않는 경우에 대해서 cnt 를 통해, 민우의 로또 번호 lottos 와 win_nums 를 리스트 형태에서 집합 형태로 형변환을 해줍니다.

 

집합 형태의 큰 장점으로는

중복을 허락하지 않고, 집합 계산(교집합, 합집합, 차집합)등의 계산에 뛰어나서 위 문제에서 민우의 번호와 로또 번호가 일치 하는지를 따로 sort 와 if 문을 통해 계산하지 않고서도 바로 계산을 가능하게 해줍니다.

 

다음으로는, 로또 번호와 일치하는 object 들에 대해서 반환한 값을 len 을 통해 길이를 구하면, 몇 개의 로또 번호와 일치하는지 여부를 cnt 를 통해 알 수 있습니다. cnt는 0~6 을 가질 수 있으며, cnt=6은 민우의 번호와 로또 번호가 모두 일치함을 의미하며, 등수를 계산하는 경우, (7- cnt) 를 통해 등수를 나타낼 수 있습니다.

 

다음으로는 0이 나오는 경우를 고려해야 합니다. 0이 나올때, 집합의 object를 변경해서 로또 번호와 같게 해주는 방식으로 해볼까했지만, 0이 있는 자리에 45가 있을 경우, 최고 등수와 최저 등수를 구할때 만약에 +1 과 같은 조건을 통해 설정한다면, 로또 번호의 길이 제한 범위 45를 초과하는 문제가 발생합니다. 그래서 단순하게, 0이 있을 경우 최고 등수 같은 경우, 로또 번호와 일치하는 경우이고, 최저는 아예 일치하지 않는 경우이므로, 단순히 0을 count 한 값을 더하거나 더하지 않는 상황을 통해, 최고 등수와 최저 등수를 구현할 수 있습니다.

 

5. 시행착오

sorted, sort 등을 통해 문제를 풀려고 하였습니다.

sort 같은 경우, 한번 실행하면, 자동으로 inplace 기능에 의해 리스트가 순서대로 변환되어집니다.

# 라이브러리
import random

num = range(0,46)
win_num = range(1,46) # 0을 포함하지 않음

# 구매한 로또 배열
lottos = random.choices(num, weights = None, k = 6)

# 당첨 번호 로또 배열
# win_nums = random.choices(win_num, weights = None, k = 6) # 같은 숫자 2개 이상 담겨있지 않음
win_nums = random.sample(win_num, 6)

print('로또 번호 : ', lottos)
print('당첨 번호 로또 배열 : ', win_nums)

# sort, 0이 있는 경우와 0이 없는 경우 나눠서 등수를 측정해야함
print('로또 번호 : ', sorted(lottos))
print('당첨 번호 로또 배열 : ', sorted(win_nums))

# cnt : 일치하는 숫자
# cnt = 6 --> 1등 
# cnt = 5 --> 2등
# cnt = 4 --> 3등
# cnt = 3 --> 4등
# cnt = 2 --> 5등
# cnt = 1 --> 6등
# cnt = 0 --> 6등 (낙첨)

# set 집합 자료형으로 풀기
# set에 다음과 같은 2가지 큰 특징
# 중복을 허용하지 않는다
# 순서가 없다(Unordered)
print(len(set(lottos) & set(win_nums)))
cnt = len(set(lottos) & set(win_nums))

print(lottos.count(0))
cnt_0 = lottos.count(0)

print('최고 등수 : ', 7-max(cnt + cnt_0,1))
print('최저 등수 : ', 7-max(cnt, 1))

로또 번호 :  [40, 11, 28, 8, 44, 29]
당첨 번호 로또 배열 :  [6, 1, 4, 31, 14, 7]
로또 번호 :  [8, 11, 28, 29, 40, 44]
당첨 번호 로또 배열 :  [1, 4, 6, 7, 14, 31]
0
0
최고 등수 :  6
최저 등수 :  6

6. 느낀점

set이라는 집합을 사용해보고, sort를 사용하지 않고도 짧게 코드 구현이 가능하다는 점이 재밌었습니다.

 

 

키워드 

알고리즘,로또,파이썬 문제,프로그래머스,python lotto, python programmers