题解 | #赏金猎人#

赏金猎人

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=' ')
全部评论

相关推荐

01-28 16:12
中南大学 Java
几年前还没有chatgpt的时候,刷题真的是很痛苦。刷不出来只能看题解,题解有几个问题:第一个是每次看的写题解的人都不一样,很难有一个统一的思路;第二个也是最重要的是,题解只提供了作者自己的思路,但是没有办法告诉你你的思路哪里错了。其实很少有错误的思路,我只是需要被引导到正确的思路上面去。所以传统题解学习起来非常困难,每次做不出来难受,找题解更难受。但是现在chatgpt能做很多!它可以这样帮助你&nbsp;-1.&nbsp;可以直接按照你喜欢的语言生成各种解法的题解和分析复杂度。2.&nbsp;把题和你写的代码都发给它,它可以告诉你&nbsp;你的思路到底哪里有问题。有时候我发现我和题解非常接近,只是有一点点🤏想错了。只要改这一点点就是最优解。信心倍增。3.&nbsp;如果遇到不懂的题解可以一行一行询问为什么要这样写,chatgpt不会嫌你烦。有时候我觉得自己的range写错了,其实那样写也没错,只是chat老师的题解有一点优化,这个它都会讲清楚。4.&nbsp;它可以帮你找可以用同类型解法来做的题。然后它可以保持解法思路不变,用一个思路爽刷一个类型的题。如果题目之间思路又有变化,它会告诉你只有哪里变了,其他的地方还是老思路。5.&nbsp;它也可以直接帮你总结模板,易错点。经过chat老师的指导,我最大的改变是敢刷题了。之前刷题需要先找某一个人写的算法题repo,然后跟着某一个人他的思路刷他给的几个题。如果想写别的题,套用思路失败了,没有他的题解,也不知道到底哪里错了;看别人的题解,思路又乱了。这个问题在二分查找和dp类型的题里面特别常见。但是现在有chat老师,他会针对我的代码告诉我我哪里想错了,应该怎么做;还按照我写代码的习惯帮我总结了一套属于我的刷题模板。每天写题全是正反馈!
牛客981:不刷才是爽
AI时代的工作 VS 传...
点赞 评论 收藏
分享
评论
1
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务