1144.递减元素使数组呈锯齿状-526互联

问题描述

给你一个整数数组 nums，每次操作会从中选择一个元素并 将该元素的值减少 1。

如果符合下列情况之一，则数组 A 就是 锯齿数组：

每个偶数索引对应的元素都大于相邻的元素，即 A[0] > A[1] < A[2] > A[3] < A[4] > ...
或者，每个奇数索引对应的元素都大于相邻的元素，即 A[0] < A[1] > A[2] < A[3] > A[4] < ...

返回将数组 nums 转换为锯齿数组所需的最小操作次数。

示例 1：

输入：nums = [1,2,3]
输出：2
解释：我们可以把 2 递减到 0，或把 3 递减到 1。

示例 2：

输入：nums = [9,6,1,6,2]
输出：4

提示：

1 <= nums.length <= 1000
1 <= nums[i] <= 1000

解题思路

首先要注意本题只允许递减数字，只需要考虑分别考虑偶数索引满足情况和奇数索引满足情况的情况，模拟即可。

代码

class Solution {
  public:
    int movesToMakeZigzag(vector<int> &nums) {
        if (nums.size() == 1)
            return 0;
        // 考虑偶数索引大的情况
        int cnt2 = 0;
        for (int i = 1; i < nums.size(); i += 2) {
            if (i + 1 < nums.size() && i - 1 >= 0) {
                int tmp = std::min(nums[i + 1], nums[i - 1]);
                if (tmp <= nums[i]) {
                    cnt2 += nums[i] - tmp + 1;
                }
            } else if (i + 1 >= nums.size()) {
                if (nums[i - 1] <= nums[i]) {
                    cnt2 += nums[i] - nums[i - 1] + 1;
                }
            } else {
                if (nums[i + 1] <= nums[i]) {
                    cnt2 += nums[i] - nums[i + 1] + 1;
                }
            }
        }
        // 考虑奇数索引大的情况
        int cnt4 = 0;
        for (int i = 0; i < nums.size(); i += 2) {
            if (i + 1 < nums.size() && i - 1 >= 0) {
                int tmp = std::min(nums[i + 1], nums[i - 1]);
                if (tmp <= nums[i]) {
                    cnt4 += nums[i] - tmp + 1;
                }
            } else if (i + 1 >= nums.size()) {
                if (nums[i - 1] <= nums[i]) {
                    cnt4 += nums[i] - nums[i - 1] + 1;
                }
            } else {
                if (nums[i + 1] <= nums[i]) {
                    cnt4 += nums[i] - nums[i + 1] + 1;
                }
            }
        }
        return std::min(cnt2, cnt4);
    }
};