LeetCode 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] 。

贪心法,遍历一遍数组,如果当前数字比前一数字小,则将其增大到前一数字加1:

class Solution {
public:
    int minOperations(vector<int>& nums) {
        int ret = 0;
        int sz = nums.size();
        int pre = nums[0];
        for (int i = 1; i < sz; ++i) {
            if (pre < nums[i]) {
                pre = nums[i];
                continue;
            }

            ret += pre - nums[i] + 1;
            ++pre;    // 当当前遍历到的值小于前一值时,将当前遍历到的值设为前一值加1
        }
        return ret;
    }
};

设输入数组长度为n,此算法时间复杂度为O(n),空间复杂度为O(1)。

posted @ 2023-01-27 15:20  epiphanyy  阅读(22)  评论(0)    收藏  举报  来源