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

https://programmers.co.kr/learn/courses/30/lessons/77484

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

import sys

input = sys.stdin.readline
lottos = list(map(int, input().split()))
win_nums = list(map(int, input().split()))


def solution(lottos, win_nums):
    if sum(lottos) == 0:
        return [1, 6]
    cnt = 0
    for i in lottos:
        if i in win_nums:
            cnt += 1
    zero = lottos.count(0)
    if cnt == 0 and zero == 0:
        return [6, 6]
    return [7 - (cnt + zero), 7 - cnt]


print(solution(lottos, win_nums))

 

해설

7 - 이미 맞은 개수가 최저 순위, 7 - 이미 맞은 개수 + 0의 개수가 최고순위다.

여기서 따로 생각할 부분은 모든 수가 0일 때와, 맞은 개수도 없고 0의 개수도 없을 때 두 가지 경우이다.

모든 수가 0이면 1등도 할 수 있고, 6등도 할 수 있다.

맞은 개수와 0의 개수가 모두 없으면 이 때는 6등밖에 할 수 없다.