Programming/Algorithm

[프로그래머스] 폰켓몬 (by Python)

oranz.23 2021. 5. 16. 03:46

문제

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

 

코딩테스트 연습 - 폰켓몬

당신은 폰켓몬을 잡기 위한 오랜 여행 끝에, 홍 박사님의 연구실에 도착했습니다. 홍 박사님은 당신에게 자신의 연구실에 있는 총 N 마리의 폰켓몬 중에서 N/2마리를 가져가도 좋다고 했습니다.

programmers.co.kr

 

풀이

def solution(nums):
    a = len(set(nums))
    b = len(nums) // 2
    return b if a > b else a

만약 폰켓몬의 종류가 가져갈 수 있는 폰켓몬의 수보다 적다면, 그 종류의 개수가 답이 될 것입니다.

그렇지 않다면, 가져갈 수 있는 폰켓몬의 수가 최댓값일 것입니다.