题解 | #赏金猎人#

赏金猎人

http://www.nowcoder.com/questionTerminal/60b84a624bee4655bce2d55087365de0

import heapq

n, k = [int(n) for n in input().split(' ')]
attack, money = [int(n) for n in input().split(' ')], [int(n) for n in input().split(' ')]

# 将 idx attack money 三个数据 绑定(使用enumerate) 并且用攻击力attack排序 从小到大
sorted_data_with_idx = sorted(enumerate(zip(attack, money)), key=lambda x: x[1][0])
#将排序号的全部解耦合 方便使用
sorted_idx, sorted_value = list(zip(*sorted_data_with_idx))
sorted_attack, sorted_money = list(zip(*sorted_value))

ans = [0]*n
#sorted_idx[0] 用来找到原来应该在的索引位置
ans[sorted_idx[0]] = sorted_money[0] #最小的只能是自己
money_list = []

#用累加和方式处理可获取的金钱
cur_sum = 0
for i in range(1, n):
    ans[sorted_idx[i]] = sorted_money[i]  #自己的money
    pre_i = i - 1
    if len(money_list) < k:
        #没到k个击杀上线 sum可以一直加
        cur_sum += sorted_money[pre_i]
        heapq.heappush(money_list, sorted_money[pre_i])
        ans[sorted_idx[i]] += cur_sum
    else:
        #到达上限了 需要选取钱多的来击杀  pushpop 保持了list_money内的数量为k  放入-排序-弹出最小 组合操作
        tmp = heapq.heappushpop(money_list, sorted_money[pre_i])
        cur_sum += sorted_money[pre_i] - tmp
        ans[sorted_idx[i]] += cur_sum
for a in ans:
    print(a, end=' ')
全部评论

相关推荐

评论
1
收藏
分享

创作者周榜

更多
牛客网
牛客企业服务