[프로그래머스] 폰켓몬

문제

https://school.programmers.co.kr/learn/courses/30/lessons/1845

프로그래머스

 

 

 

---

 

주요 아이디어

문제 길이는 상당히 길지만 set을 쓰면 굉장히 간단하게 풀리는 문제이다.

N/2마리의 폰켓몬을 선택하는 방법 중 가장 많은 종류의 폰켓몬을 선택하려면, 다음 조건문을 만족하면 된다.

• 전달 받은 폰켓몬 종류 번호의 unique 값이 N/2일 경우 : 그대로 N/2 반환
• 전달 받은 폰켓몬 종류 번호의 unique 값이 N/2 보다 작을 경우 : unique 값 반환

 

---

 

코드 구현 (Python 3)

 

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

 

 

---

 

제출 결과