[Programmers/알고리즘 고득점 KIT] 포켓몬

[Programmers] 포켓몬

Level 1 | #해시

문제 바로가기 🔗

1. 문제 요약

N마리의 포켓몬 중에서 N/2마리를 가져갈 수 있다. 이때 최대한 다양한 종류의 포켓몬을 선택하는 것이 목표이다.

포켓몬의 종류 번호가 담긴 배열 nums가 주어질 때, 선택할 수 있는 포켓몬 종류 개수의 최댓값을 구하는 문제이다.

2. 접근 방식 및 포인트

가장 많은 종류를 선택하려면 중복된 종류를 제거한 뒤, 내가 가져갈 수 있는 수(N/2)와 비교해야 한다.

• ✅ 자료구조: 중복을 허용하지 않는 set을 사용하여 포켓몬의 종류를 파악한다.
• ✅ 로직: 포켓몬 종류의 수(len(set(nums)))가 내가 가질 수 있는 수(len(nums) // 2)보다 크면 N/2를 반환하고, 작으면 종류의 수를 그대로 반환한다.

3. 구현 코드 (Python)

def solution(nums):
    s = set(nums)   # 중복 제거
    if len(s) > len(nums) // 2:
        return len(nums) // 2
    else:
        return len(s)

📌 중복 제거하는 방법: set()에 담으면 자동으로 중복이 제거된다.

📌 set의 특징: iterable(리스트, 튜플 등)이면 변환이 가능하지만, 원소는 반드시 hashable(불변 타입)이어야 한다.

4. 리팩토링 - 한 번에 리턴

min() 함수를 사용하면 조건문 없이 더 간결하게 작성이 가능하다.

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

포켓몬 종류의 수와 가질 수 있는 최대 수(N/2) 중에서 더 작은 값을 선택하는 것이 곧 문제의 정답이 된다. 가독성과 간결함 면에서 훨씬 효율적인 코드이다.