프로그래머스 폰켄몬(파이썬)

최대한 다양하게 고르라는 것을 보고 바로 list(set())이 떠올랐다.

def solution(nums):
    answer = 0
    a=list(set(nums))   # set로 중복값 제거
    if len(a)>len(nums)//2:     #하지만 a의 개수가 고를수 있는 n/2마리보다 클 경우
        answer = len(nums)//2   #n/2개만 선택
    else:
        answer=len(a)   #아니면 전부 선택
    return answer