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

题目分析

1. 每次操作只是对一个元素减1
2. 操作时取前后的最大差值并加1
3. 分别对偶数位和奇数位进行操作并求和，取其最小值。

    int movesToMakeZigzag(vector& nums) {
        int ans[2];
        ans[1] = ans[0] = 0;
        for(int i = 0; i < nums.size(); i++) {
            int d1 =0,d2=0;
            //对前一个元素判别需要减去多少
            if(i > 0 && nums[i] >= nums[i-1] ) {
                d1 = nums[i] - nums[i-1] + 1;
            }
            //对后一个元素判别需要减去多少
            if(i < nums.size()-1 && nums[i] >= nums[i+1]) {
                d2 = nums[i] - nums[i+1] + 1;
            }
            //取前后最大值并求和。
            ans[i%2] += max(d1,d2);
        }
        return min(ans[1], ans[0]);
    }