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

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

坚持刷leetcode,巩固算法基础。 这个题目刚看的时候感觉稍微有点复杂,想了一下其实没有什么优化问题,分下面两种情况讨论,数组元素一个一个的看,遇到不符合要求的就直接改掉,相当于一个贪心。

给你一个整数数组 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;
        vector<int>a1(nums);
        vector<int>a2(nums);
        int counter1 = 0;
        int counter2 = 0;
        for(int i=0;i<nums.size()-1;i++)
        {
            if(i%2==0)
            {
                if(a1[i]<a1[i+1])
                    continue;
                else
                {
                    counter1 = counter1 + a1[i]-a1[i+1]+1;
                    a1[i]=a1[i+1]-1;
                }
            }
            else{
                if(a1[i]>a1[i+1])
                    continue;
                else
                {
                    counter1 = counter1 + a1[i+1]-a1[i]+1;
                    a1[i+1] = a1[i]-1;
                    
                }
            }
            
        }
        
        for(int i=0;i<nums.size()-1;i++)
        {
            if(i%2!=0)
            {
                if(a2[i]<a2[i+1])
                    continue;
                else
                {
                    counter2 = counter2 + a2[i]-a2[i+1]+1;
                    a2[i]=a2[i+1]-1;
                }
            }
            else{
                if(a2[i]>a2[i+1])
                    continue;
                else
                {
                    counter2 = counter2 + a2[i+1]-a2[i]+1;
                    a2[i+1] = a2[i]-1;
                    
                }
            }
            
        }
        return min(counter1, counter2);
        
    }
};
发布了85 篇原创文章 · 获赞 100 · 访问量 13万+

猜你喜欢

转载自blog.csdn.net/caozixuan98724/article/details/98785353