I about me

[Python] 폰켓몬 본문

Algorithm/프로그래머스

[Python] 폰켓몬

ssungni 2024. 6. 24. 13:45

 

예시 분석 

가장 많은 종류의 폰켓몬을 선택하는 방법을 찾아, 그때의 폰켓몬 종류 번호의 개수를 return하도록 solution 함수를 완성하여라.

 

1. [3, 1, 2, 3] → 일단 n/2 = 2개를 가져갈 수 있음, 가장 많은 종류로도 2개 선택가능

2. [3, 3, 3, 2, 2, 4] → 일단  n/2 =  3개를 가져갈 수 있음,  가장 많은 종류로도 3개 선택 가능

3. [3, 3, 3, 2, 2, 2] → 일단  n/2 =  3개를 가져갈 수 있음, 그러나 가장 많은 종류로는 2개 선택 가능함

 

풀이 1

def solution(nums):
    if (len(nums)/2) <= len(set(nums)):
        return (len(nums)/2)
    return len(set(nums))

 

풀이 2

def solution(nums):
    return min(len(nums)/2, len(set(nums)))