Leetcode 5147：递减元素使数组呈锯齿状

题目描述

给你一个整数数组 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) {
        nums.push_back(INT_MAX);
        int ans = INT_MAX,len = nums.size();
        int tans = 0;
        //偶数位置大
        for(int i=1;i+1<len;i+=2){
            int tmp = min(nums[i-1],nums[i+1]);
            if(nums[i]>=tmp) tans += nums[i]-tmp+1;
        }
        ans = min(ans,tans);
        tans = 0;
        //奇数位置大
        for(int i=1;i+1<len;i+=2){
            if(nums[i]<=nums[i-1]) tans += nums[i-1]-nums[i]+1;
            if(nums[i+1]!=INT_MAX && nums[i]<=nums[i+1]){
                tans += nums[i+1]-nums[i]+1;
                nums[i+1] = nums[i]-1;
            }
        }
        ans = min(ans,tans);
        return ans;
    }
};