-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathmaxJumps.py
30 lines (26 loc) · 946 Bytes
/
maxJumps.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
class Solution:
def maxJumps(self, nums: List[int], d: int) -> int:
N, dp, heap = len(nums), [1] * len(nums), []
for idx, num in enumerate(nums):
heappush(heap, (num, idx))
while heap:
num, i = heappop(heap)
cur_max = 0
for delta in range (1, d + 1):
j = i + delta
if j >= N: break
cur_max = max(cur_max, nums[j])
if nums[i] > cur_max:
dp[i] = max(dp[i], 1 + dp[j])
else:
break
cur_max = 0
for delta in range (1, d + 1):
j = i - delta
if j < 0: break
cur_max = max(cur_max, nums[j])
if nums[i] > cur_max:
dp[i] = max(dp[i], 1 + dp[j])
else:
break
return max(dp)