[프로그래머스] 폰켓몬 Python 3 Code

이번에도 프로그래머스 스쿨 추천 문제 중 하나인

폰켓몬을 풀어보겠다.

---

문제

---

예시

 

---

Code

 

구현은 중복을 지우는 set를 써서 포켓몬 종류를 구한 뒤

len과 비교하여 더 작은 것을 정답으로 리턴하게 했다.

 

def solution(nums):
    answer = 0
    if(len(nums)/2) > len(set(nums)):
        answer = len(set(nums))
    else:
        answer = len(nums)/2

    return answer

간결하게 짰다고 생각했는데 역시 파이썬은 기능이 많은 것을 활용해야 한다는 것을 아래 코드에서 느꼈다.

 

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

min을 사용하면 이렇게 한 줄로 구현할 수 있다.