프로그래머스 1단계 "폰켓몬"

문제

프로그래머스 1단계 폰켓몬

풀이

처음에 문제를 봤을 땐 조합문제인가 싶었는데 생각해보면 간단한 문제이다.
N개의 포켓몬의 종류를 파악하고(코드에선 set()으로 중복을 제거), 포켓몬 수 N의 절반보다 종류수가 적으면 종류수가 정답이 되고, 종류수가 N/2보다 많으면 N/2가 정답이 된다.

Python 코드

import sys
input = sys.stdin.readline

def solution(nums):
  a = len(set(nums))

  if len(nums)//2 > a:
    return a
  else:
    return len(nums)//2