做项目的最大收益问题

来个Python3版本的

N, W, K = (int(_) for _ in input().strip().split())
if N <=0:
    print(W)
costs = [int(_) for _ in input().strip().split()]
profits = [int(_) for _ in input().strip().split()]
items = [(a, b) for a, b in zip(costs, profits)]
import heapq
heapq.heapify(items)  # 先把所有项目按照花费的小根堆组织起来
more_q = []  # 大根堆按照profits组织
# 循环 K 次选择 K 个项目
for i in range(K):
    # 把当前资金能做的项目从小根堆中弹出，并进入大根堆
    while len(items)>0 and items[0][0] <= W:
        one_item = heapq.heappop(items)
        heapq.heappush(
            more_q, (-one_item[1],) + one_item
        )  # 由于默认小根堆，所以把利润取反后push进小根堆，就是大根堆了
    # 循环结束时， one_item 是当前轮初始资金不能做的
    if len(more_q) == 0: # 如果大根堆没有元素了，代表当前的资金无法做任何项目了，可以提前停止了
        break

    # 从大根堆中取出一个项目做，并把收益加到资金中
    do_item = heapq.heappop(more_q)[1:]
    W +=  do_item[1]
print(W)