[프로그래머스] 포켓몬 - Python

1. 문제 설명

• 입력 nums : 포켓몬 종류를 담은 배열
• 출력 주어진 배열의 길이의 절반만큼 수를 선택할 수 있고, 최대한 다양한 종류의 수를 선택해야할 때 선택할 수 있는 최대값을 return

nums	result
[3,1,2,3]	2
[3,3,3,2,2,4]	3
[3,3,3,2,2,2]	2

2. 코드

def solution(nums):
    answer = 0
    max_num = len(nums)/2
    temp = [] #지금까지 나온 포켓몬 저장
    for i in range(len(nums)):
        if nums[i] not in temp:
            temp.append(nums[i])
        if len(temp) == max_num:
            break
    return len(temp)

3. 어려웠거나 배운 점

not in 활용

not in을 쓰지 않는다면 코드가 길어질 것

특정 값이 리스트에 존재하는지 검사할 때, in 뿐만아니라 not in 도 기억할 것