[Python] 폰켓몬

문제 참고

프로그래머스

 

코드 작성

def solution(nums):
    answer = 0
    num_set = list(set(nums))
    half_num = int(len(nums) / 2)
    
    if len(num_set) > half_num:
        answer = half_num
    else:
        answer = len(num_set)
    return answer

- nums에 중복되는 폰켓몬을 제거하여 num_set에 저장

- half_num은 최대로 가져갈 수 있는 폰켓몬의 수이다.

- 만약 nums의 중복이 적어 half_num보다 많은 경우 정답은 half_num

- nums의 중복이 많아서 half_num보다 작은 경우 num_set이 정답