题解 | #赏金猎人#
赏金猎人
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=' ')