[프로그래머스/알고리즘] 폰켓몬 [파이썬/해시]

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

프로그래머스

간단한 문제였다.

그런데 처음에 조합이랑 카운터를 사용했는데 시간초과가 났다.

보니까 set을 이용하면 반복문없이 길이 비교만 하면 됐다,,

 

 

시간초과 코드

from itertools import combinations
from collections import Counter
def solution(nums):
    answer = 0
    take = int(len(nums)/2)
    for i in combinations(nums,take):
        c = Counter(i)
        answer = max(len(c.keys()), answer)
        
        
    return answer

정답코드 set 이용

def solution(nums):
    answer = 0
    take = int(len(nums)/2)
    s = set(nums)
    if len(s)<=take:
        answer = len(s)
    else:
        answer = take
        
return answer