题目描述
给你一个整数数组 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
解释:你可以进行如下操作:
- 增加 nums[2] ,数组变为 [1,1,2] 。
- 增加 nums[1] ,数组变为 [1,2,2] 。
- 增加 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
思路
- 初始化mx=0,mx存储当前最大值。初始化ret=0,ret是使 nums 严格递增 的最少操作次数
- 从左向右遍历数组
- 如果当前位num<mx+1
- 更新ret+=max(0,mx+1-num)
- 更新mx=mx+1
- 如果当前位num不小于mx+1
- 更新mx=num
Java实现
class Solution {
public int minOperations(int[] nums) {
// 1. 初始化mx=0,mx存储当前最大值。初始化ret=0,ret是使 nums 严格递增 的最少操作次数
int mx=0;
int ret=0;
// 2. 从左向右遍历数组
for(int num:nums)
{
// 3. 如果当前位num<mx+1
if(num<mx+1)
{
// 4. 更新ret+=max(0,mx+1-num)
ret += Math.max(0,mx+1-num);
// 5. 更新mx=mx+1
mx=mx+1;
}
// 6. 如果当前位num不小于mx+1
else
{
// 7. 更新mx=num
mx=num;
}
}
return ret;
}
}
python实现
class Solution:
def minOperations(self, nums: List[int]) -> int:
# 1. 初始化mx=0,mx存储当前最大值。初始化ret=0,ret是使 nums 严格递增 的最少操作次数
mx,ret=0,0
# 2. 从左向右遍历数组
for num in nums:
# 3. 如果当前位num<mx+1
if num<mx+1:
# 4. 更新ret+=max(0,mx+1-num)
ret+=max(0,mx+1-num)
# 5. 更新mx=mx+1
mx=mx+1
# 6. 如果当前位num不小于mx+1
else:
# 7. 更新mx=num
mx=num
return ret
C++实现
class Solution {
public:
int minOperations(vector<int>& nums) {
// 1. 初始化mx=0,mx存储当前最大值。初始化ret=0,ret是使 nums 严格递增 的最少操作次数
int mx=0;
int ret=0;
// 2. 从左向右遍历数组
for(int num:nums)
{
3. 如果当前位num<mx+1
if(num<mx+1)
{
// 4. 更新ret+=max(0,mx+1-num)
ret+=max(0,mx+1-num);
// 5. 更新mx=mx+1
mx=mx+1;
}
// 6. 如果当前位num不小于mx+1
else
{
// 7. 更新mx=num
mx=num;
}
}
return ret;
}
};