leetcode之1144. 递减元素使数组呈锯齿状

题目

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

题解

修改当前位置左右两边的值，通过第一次传入的是0,还是1控制是奇数还是偶数索引

代码

class Solution {
public:
    size_t len;
    
    int compute(vector<int>& arr, size_t idx) {
        int count = 0;
        for (; idx < len; idx += 2) {
            int cur_num = arr[idx];
            if (idx > 0 && arr[idx - 1] >= cur_num) {
                count += arr[idx - 1] - cur_num + 1;
                arr[idx - 1] = cur_num - 1;
            }
            if (idx + 1 < len && arr[idx + 1] >= cur_num) {
                count += arr[idx + 1] - cur_num + 1;
                arr[idx + 1] = cur_num - 1;
            }
        }
        return count;
    }
    
    int movesToMakeZigzag(vector<int>& nums) {
        len = nums.size();
        vector<int> copy_nums(nums);
        return min(compute(copy_nums, 0), compute(nums, 1));
    }
};