[프로그래머스]폰켓몬(Python)

0. 문제 링크

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

프로그래머스

 

1. 문제 설명

1) 주어진 배열에서 폰켓몬의 종류 정리하기

2) N 마리 폰켓몬 중 가장 많은 종류의 폰켓몬 N // 2마리 고르기

 

2. 입출력

# input
nums = [3, 1, 2, 3]

# output
result = 2

 

3. 코드

1) 첫 번째 코드

def solution(nums):
    answer = 0
    kemon = []

    for i in range(len(nums)):
        if nums[i] not in kemon:    # kemon에 nums[i]가 없으면
            kemon.append(nums[i])   # nums[i] 추가

    answer = len(kemon) if len(kemon) < len(nums)//2 else len(nums)//2
    
    return answer

실행 결과

kemon 리스트에 nums[i]가 있는지 반복적으로 확인하다보니

테스트 16, 17, 18에서는 시간이 오래 걸림

 

2) 두 번째 코드

def solution(nums):
    answer = 0

    answer = min(len(set(nums)), len(nums)//2) # 주어진 배열의 종류나 //2 중 작은 값 선택
    return answer

 

실행 결과

배열의 종류만 세고 min 함수로 처리하다보니 테스트 14, 15, 16에서도 비교적 시간이 단축됨

 

4. 알게된 점

1) set 함수의 사용

중복을 허용하지 않고 순서가 없기 때문에 종류를 셀 때 유리함

s1 = set([1, 2, 2, 3])
s1

# {1, 2, 3}

 

5. Github

https://github.com/iihye/Algorithm/blob/main/Programmers/pon_kemon.py

GitHub - iihye/Algorithm