题目链接
Leetcode.2383 赢得比赛需要的最少训练时长 Rating : 1413
题目描述
你正在参加一场比赛,给你两个 正 整数 initialEnergy
和 initialExperience
分别表示你的初始精力和初始经验。
另给你两个下标从 0 开始的整数数组 energy
和 experience
,长度均为 n 。
你将会 依次 对上 n 个对手。第 i
个对手的精力和经验分别用 energy[i]
和 experience[i]
表示。当你对上对手时,需要在经验和精力上都 严格 超过对手才能击败他们,然后在可能的情况下继续对上下一个对手。
击败第 i
个对手会使你的经验 增加 experience[i]
,但会将你的精力 减少 energy[i]
。
在开始比赛前,你可以训练几个小时。每训练一个小时,你可以选择将增加经验增加 1
或者 将精力增加 1
。
返回击败全部 n 个对手需要训练的 最少 小时数目。
示例 1:
输入:initialEnergy = 5, initialExperience = 3, energy = [1,4,3,2], experience = [2,6,3,1]
输出:8
解释:在 6 小时训练后,你可以将精力提高到 11 ,并且再训练 2 个小时将经验提高到 5 。
按以下顺序与对手比赛:
-你的精力与经验都超过第 0 个对手,所以获胜。
精力变为:11 - 1 = 10 ,经验变为:5 + 2 = 7 。
你的精力与经验都超过第 1 个对手,所以获胜。
精力变为:10 - 4 = 6 ,经验变为:7 + 6 = 13 。
你的精力与经验都超过第 2 个对手,所以获胜。
精力变为:6 - 3 = 3 ,经验变为:13 + 3 = 16 。
你的精力与经验都超过第 3 个对手,所以获胜。
精力变为:3 - 2 = 1 ,经验变为:16 + 1 = 17 。
在比赛前进行了 8 小时训练,所以返回 8 。
可以证明不存在更小的答案。
示例 2:
输入:initialEnergy = 2, initialExperience = 4, energy = [1], experience = [3]
输出:0
解释:你不需要额外的精力和经验就可以赢得比赛,所以返回 0 。
提示:
- n = = e n e r g y . l e n g t h = = e x p e r i e n c e . l e n g t h n == energy.length == experience.length n==energy.length==experience.length
- 1 < = n < = 100 1 <= n <= 100 1<=n<=100
- 1 < = i n i t i a l E n e r g y , i n i t i a l E x p e r i e n c e , e n e r g y [ i ] , e x p e r i e n c e [ i ] < = 100 1 <= initialEnergy, initialExperience, energy[i], experience[i] <= 100 1<=initialEnergy,initialExperience,energy[i],experience[i]<=100
解法:贪心
我们分别用 a
和b
来表示,需要额外进行训练的 精力 和 经验。
精力:
因为每次打败一个对手 都要用自己的精力减去对手的精力。所以初始的精力应该是 大于 所有对手的精力的和, i n i t i a l E n e r g y > s u m ( e n e r g y ) initialEnergy > sum(energy) initialEnergy>sum(energy),即 i n i t i a l E n e r g y = s u m ( e n e r g y ) + 1 initialEnergy = sum(energy) + 1 initialEnergy=sum(energy)+1。
- 如果
i
n
i
t
i
a
l
E
n
e
r
g
y
>
s
u
m
(
e
n
e
r
g
y
)
initialEnergy > sum(energy)
initialEnergy>sum(energy),
a = 0
。 - 如果
i
n
i
t
i
a
l
E
n
e
r
g
y
≤
s
u
m
(
e
n
e
r
g
y
)
initialEnergy \leq sum(energy)
initialEnergy≤sum(energy),
a = sum(energy) - initialEnergy + 1
。
经验:
因为每次打败一个对手 都要加上对手的经验。所以,我们只能从第一个对手开始进行模拟,对于当前经验 大于对手的经验 和 小于等于对手的经验 分别处理。
-
i
n
i
t
i
a
l
E
x
p
e
r
i
e
n
c
e
>
e
x
p
e
r
i
e
n
c
e
[
i
]
initialExperience > experience[i]
initialExperience>experience[i],
initialExperience += experience[i]
。 -
i
n
i
t
i
a
l
E
x
p
e
r
i
e
n
c
e
<
=
e
x
p
e
r
i
e
n
c
e
[
i
]
initialExperience <= experience[i]
initialExperience<=experience[i],
b += experience[i] - initialExperience + 1
,initialExperience = experience[i] + 1
,initialExperience += experience[i]
。
时间复杂度: O ( n ) O(n) O(n)
C++代码:
class Solution {
public:
int minNumberOfHours(int initialEnergy, int initialExperience, vector<int>& energy, vector<int>& experience) {
int sum = accumulate(energy.begin(),energy.end(),0);
int a = max(0 , sum - initialEnergy + 1);
int b = 0;
int n = experience.size();
for(auto e:experience){
if(initialExperience <= e){
b += e - initialExperience + 1;
initialExperience = e + 1;
}
initialExperience += e;
}
return a + b;
}
};
Python代码:
class Solution:
def minNumberOfHours(self, initialEnergy: int, initialExperience: int, energy: List[int], experience: List[int]) -> int:
a = max(0 , sum(energy) - initialEnergy + 1)
b = 0
for e in experience:
if initialExperience <= e:
b += e - initialExperience + 1
initialExperience = e + 1
initialExperience += e
return a + b