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

2021 dev-matching에 나온 문제인데, 대회 당시에는 하나의 케이스로 인해 통과하지 못했다. 

14번 케이스가 문제인데, 우선 오답코드를 살펴보자.

 

오답 코드

def solution(lottos, win_nums):
    count_zero = lottos.count(0)
    count = [1 for num in lottos if num in win_nums]

    if not count:
        return [1, 6]

    return [7 - (len(count) + count_zero), 7 - len(count)]

 하나도 맞은게 없다는 것(count 가 0)은 2가지로 분류 되는데,

1. lottos가 모두 0일 땐 [1,6]이 되고,
2. lottos에 0이 하나도 없을 땐, [6,6]이 된다.

이 점을 간과하여 14번 케이스를 뚫지 못했다. (1번 케이스만 생각한거지)

 

 

그렇기 때문에 2번의 경우를 고려하여 예외처리를 해주면 통과할 수 있다.

정답 코드

def solution(lottos, win_nums):
    count_zero = lottos.count(0)
    count = [1 for num in lottos if num in win_nums]
    if (count_zero) == 6:
        return [1, 6]
    elif not count:
        return [6, 6]
    else:
        return [7 - (len(count) + count_zero), 7 - len(count)]

 

 

다른 사람 풀이 

훨씬 직관적이고, lottos를 win_nums에 맞추는 것이 아닌 그 역으로 접근했다.

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]