강의로 돌아가기
-

파이썬 이거보다 줄일 수 있어요..???

def solution(n, works):
    if sum(works)<=n:return 0
    while n:
        works[works.index(max(works))]-=1
        n-=1
    return sum([i*i for i in works])

효율성테스트 IBM컴퓨터 기준인가요.. 더 줄이는 법 아시는 분..?

  • Danggai

    코드가 짧다고 효율적인게 아니에요;;

    Danggai―2019.12.18 13:41
  • MH_K

    heapq모듈 써보세요! 최대힙으로 했을때 통과가 되네요!

    MH_K―2020.01.02 17:55
2 개의 답변
황규승

binary_search 이용하면 됩니다.

맹산하

최대힙을 사용하시면 됩니다.

답변 쓰기
이 입력폼은 마크다운 문법을 지원합니다. 마크다운 가이드 를 참고하세요.