给你一个整数数组 nums
(下标从 0 开始)。每一次操作中,你可以选择数组中一个元素,并将它增加 1
。
- 比方说,如果
nums = [1,2,3]
,你可以选择增加nums[1]
得到nums = [1,3,3]
。
请你返回使 nums
严格递增 的 最少 操作次数。
我们称数组 nums
是 严格递增的 ,当它满足对于所有的 0 <= i < nums.length - 1
都有 nums[i] < nums[i+1]
。一个长度为 1
的数组是严格递增的一种特殊情况。
示例 1:
输入:nums = [1,1,1] 输出:3 解释:你可以进行如下操作: 1) 增加 nums[2] ,数组变为 [1,1,2] 。 2) 增加 nums[1] ,数组变为 [1,2,2] 。 3) 增加 nums[2] ,数组变为 [1,2,3] 。
示例 2:
输入:nums = [1,5,2,4,1] 输出:14
示例 3:
输入:nums = [8] 输出:0
提示:
1 <= nums.length <= 5000
1 <= nums[i] <= 104
遍历数组,维护一个 preMax 变量。
class Solution:
def minOperations(self, nums: List[int]) -> int:
n = len(nums)
pre_max = nums[0]
times = 0
for i in range(1, n):
if nums[i] <= pre_max:
steps = pre_max - nums[i] + 1
times += steps
pre_max = nums[i] + steps
else:
pre_max = nums[i]
return times
class Solution {
public int minOperations(int[] nums) {
int n = nums.length;
int preMax = nums[0];
int times = 0;
for (int i = 1; i < n; ++i) {
if (nums[i] <= preMax) {
int steps = preMax - nums[i] + 1;
times += steps;
preMax = nums[i] + steps;
} else {
preMax = nums[i];
}
}
return times;
}
}