강의로 돌아가기
강민웅

효율성에 test3번을 알고 싶은데요

자꾸 거기서만 문제가 발생하네요....

작성중인 코드―solution.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
import heapq
def solution(stock, dates, supplies, k):
    answer = 0
    h = []
    idx = 0
    if stock >= k:
        return 0
    else:
        for i in range(0, k):
            if idx < len(dates) and i == dates[idx]:
                heapq.heappush(h, -supplies[idx])
                idx += 1

            if stock == 0:
                stock += -heapq.heappop(h)
                answer += 1
                if stock >= k-1:
                    return answer         

            stock -=1
    return answer
0 개의 답변
답변 쓰기
이 입력폼은 마크다운 문법을 지원합니다. 마크다운 가이드 를 참고하세요.