본문 바로가기

Algorithm/Practice

2021 Dev-Matching: 웹 백엔드 개발(상반기) #1

로또의 최고순위와 최저순위


Level 1

 

 

코딩테스트 연습 - 로또의 최고 순위와 최저 순위

로또 6/45(이하 '로또'로 표기)는 1부터 45까지의 숫자 중 6개를 찍어서 맞히는 대표적인 복권입니다. 아래는 로또의 순위를 정하는 방식입니다. 1 순위 당첨 내용 1 6개 번호가 모두 일치 2 5개 번호

programmers.co.kr

 

문제 설명

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

순위당첨 내용

 로또를 구매한 민우는 당첨 번호 발표일을 학수고대하고 있었습니다. 하지만, 민우의 동생이 로또에 낙서를 하여, 일부 번호를 알아볼 수 없게 되었습니다. 당첨 번호 발표 후, 민우는 자신이 구매했던 로또로 당첨이 가능했던 최고 순위와 최저 순위를 알아보고 싶어 졌습니다.
 알아볼 수 없는 번호를 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 함수를 완성해주세요.

 

Logic

 문제가 하라는 대로 푸는 구현 문제. 일치하는 개수, 0의 개수를 세서 순위를 계산하면 된다. 중복되는 숫자가 없으며, 주의할 점은 테스트케이스에 나오지 않은 '0의 개수가 없고, lottos와 win_nums 의 원소가 모두 일치하지 않을때' 만 신경쓰면 되었다.

 

Tip

- .count(특정 원소) : 파이썬 리스트에서 특정 원소의 개수를 세는 함수 

- 결과 A if 조건 else 결과 B : 파이썬의 삼항 연산자

 

Code

def solution(lottos, win_nums):
    
    same_cnt = 0 # 당첨번호와 일치하는 값 개수
    unknown_cnt = lottos.count(0) # 알 수 없는 값 개수
    
    for i in range(len(lottos)):
        if lottos[i]!= 0 and lottos[i] in win_nums:
            same_cnt += 1
    
    if unknown_cnt == 0:
        lowest = highest = 6 - same_cnt + 1
    else:
        lowest = 6 - same_cnt + 1
        highest = 6 - same_cnt - unknown_cnt + 1
    lowest = 6 if lowest > 6 else lowest
    highest = 6 if highest > 6 else highest
    
    return highest, lowest

 

Simple Code1

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]

 다른 분의 짧은 풀이ㅋㅋㅋ. 역시 0의 개수를 찾고, 일치하는 값을 찾는다. 다만 순위를 구하는 방식이 멋있다. 일치하는 개수와 순위는 반비례 관계인 점과 rank를 리스트로 만들어 사용했다. 이렇게 풀면 모두 0이 아니고 일치하지 않는 경우의 로직을 따로 작성하지 않아도 된다. 그리고 순위를 리스트에 append 안하고 쉼표로 리턴해도 되는 걸 알게 되었다.

최고 순위 : rank[0의 개수 + 일치하는 개수] 

최저 순위 : rank[일치하는 개수]

Simple Code2

def solution(lottos, win_nums):
    rank = {
        0: 6,
        1: 6,
        2: 5,
        3: 4,
        4: 3,
        5: 2,
        6: 1
    }
    return [rank[len(set(lottos) & set(win_nums)) + lottos.count(0)], rank[len(set(lottos) & set(win_nums))]]

ㅋㅋㅋㅋ 내코드 눈감아. set을 이용해서 일치하는 값을 찾는 풀이다. 우와~