1827. 最少操作使数组递增

题目描述

给你一个整数数组 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

思路

1. 初始化mx=0，mx存储当前最大值。初始化ret=0，ret是使 nums 严格递增 的最少操作次数
2. 从左向右遍历数组
3. 如果当前位num<mx+1
4. 更新ret+=max(0，mx+1-num)
5. 更新mx=mx+1
6. 如果当前位num不小于mx+1
7. 更新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;
    }
};