【备战秋招】每日一题:塔子大厂真题模拟赛-第一题-魔法石(Ⅰ)

为了更好的阅读体检，可以查看我的算法学习网
本题在线评测链接:P1211

题目内容

塔子哥有$n$个魔法石，初始时魔法石的价值为$a_i$.塔子哥还有$m$点能量。对于每个能量，它可以用来点亮一颗魔法石。第$i$个魔法石被第一次点亮后的价值为$b_i$.请问塔子哥该如何安排手中的能量，使得魔法石的价值总和最大呢？

输入描述

第一行两个整数$n$,$m$ ($1\le n\le 1e5,0\le m\le n$) , 分别代表魔法石的个数以及能量数量.

第二行$n$个整数:$a_1,a_2,...,a_n$ , 代表这$n$个魔法石初始价值

第三行$n$个整数:$b_1,b_2,...,b_n$ , 代表这$n$个魔法石被点亮后的价值

$1\le a_i,b_i\le 1e9$

输出描述

一行整数，代表最大价值。

样例1

输入

3 3
1 2 3
3 2 1

输出

8

样例2

输入

3 0
1 2 3
3 2 1

输出

6

思路:贪心

$m=0$，那么就是全选$a_i$。

$m=1$,可以视作先全选$a_i$ , 然后再选一个 ${\Delta }_i=b_i-a_i$ 最大的

$m>1$ ,显然是全选$a_i$ 然后加上前$m$大的${\Delta }_i$ 。注意可能存在${\Delta }_i<0$的情况，需要忽略。这种就是我们不必用完$m$次机会的情况。

做法:先将$a_i$求和。然后对${\Delta }_i=b_i-a_i$降序排序，求和前$m$大的${\Delta }_i$。

时间复杂度

排序:$O(nlogn)$

代码

python

n , m = list(map(int , input().split()))
a = list(map(int , input().split()))
b = list(map(int , input().split()))
c = [b[i] - a[i] for i in range(n)]
c.sort()
ans = sum(a)
for i in range(n - 1 , n - 1 - m , -1):
	if c[i] > 0:
		ans += c[i]
print(ans)