[파이썬] 프로그래머스 귤 고르기

시간 초과가 나서 set을 일일히 구현하는 방식으로 바꾼 후, list 내의 count 역시 직접 구현을 했더니 시간 초과가 나지 않았다.

 

 

def solution(k, a):
    answer = 0
 
    a=list(sorted(a))
 
    gesu = [ ]
 
    count = 1
 
    for i in range(0,len(a)-1):
        if a[i]==a[i+1]:
            count+=1
        else:
            gesu.append(count)
            count=1
 
    gesu.append(len(a)-sum(gesu))
 
    gesu = list(sorted(gesu,reverse=True))
 
    for i in range(0,len(gesu)):
        k = k-gesu[i]
        answer+=1
 
        if k<=0:
            break
 
 
    return answer