[프로그래머스] 폰켓몬 #해시 #알고리즘

문제 바로가기 👉 프로그래머스

---

☑️  문제 요약

당신은 폰켓몬을 잡기 위한 오랜 여행 끝에, 홍 박사님의 연구실에 도착했습니다. 

홍 박사님은 당신에게 자신의 연구실에 있는 총 N 마리의 폰켓몬 중에서 N/2마리를 가져가도 좋다고 했습니다.

1. N마리의 폰켓몬 중 N/2마리를 선택할 때, 최대한 다양한 종류를 선택해야 한다.
2. 같은 번호의 폰켓몬은 같은 종류로 간주된다.
3. 선택할 수 있는 폰켓몬의 최대 종류 개수를 반환해야 한다.
4. nums 배열의 길이 N은 1 이상 10,000 이하의 짝수이며, 폰켓몬 번호는 1 이상 200,000 이하이다.
5. 최적의 선택 방법이 여러 개여도, 최대 종류 개수 하나만 반환한다.

 

☑️  KEY 포인트

1. 동일한 배열에서 같은 폰켓몬 종류를 제외한, Unique한 종류 개수를 구해야 한다. 따라서 list를 set으로 변환 시켜야 한다.
2. Unique한 종류 개수와 선택해야 하는 폰켓몬 수(N/2)를 비교하여 최대 종류 개수를 구할 수 있다.
3. 따라서 2가지 경우가 발생하고, 조건문을 통해 2가지 경우의 수에 해당하는 결과를 반환시킨다.
   • 배열의 unique 종류 개수  >  선택해야 하는 폰켓몬 수(N/2)
   • 배열의 unique 종류 개수  <=  선택해야 하는 폰켓몬 수(N/2)
4. 최대 종류 개수를 구해야하므로, 두 수가 같은 경우는 배열의 unique 종류 개수가 되어야 한다.

 

☑️  해결 코드

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