[프로그래머스] 폰켓몬(파이썬)

코딩테스트 연습 - 폰켓몬 | 프로그래머스 스쿨 (programmers.co.kr)

프로그래머스

 

[문제 요약]

- nums 리스트에 포켓몬 번호가 주어진다.

- 포켓몬은 N/2 마리만 가질 수 있다.

- 나는 최대한 많은 종류의 포켓몬을 가지고 싶다.

- 가질 수 있는 가장 많은 가짓수를 출력한다.

 

[문제 해결 과정]

- 최대한 많은 종류기 때문에 중복은 제거한다. -> set 이용

- N/2 이기 때문에 2로 나눠준다.

- 1번 예시처럼 포켓몬의 종류가 nums/2보다 큰 경우엔 무조건 리스트 길이를 2로 나눈 값이 정답이다.

 

[정답]

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