LeetCode-Python-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 <= nums.length <= 1000
• 1 <= nums[i] <= 1000

思路：

题目一开始翻译有问题，把我搞晕了……

做法还是比较简单的，分别处理两种情况，

大体上是按要求把应该更小的数变成它左右邻居较小值 - 1，

全部处理完之后，统计一下跟原数组的差是多少，这个差就是这种情况需要的操作数。

class Solution(object):
    def movesToMakeZigzag(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        backup = nums[:]
        cnt = 0
        for i in range(1, len(nums), 2): #保证偶数索引更大，就要让奇数索引变小
            if i != len(nums) - 1:
                if nums[i] >= nums[i - 1] or nums[i] >= nums[i + 1]:
                    nums[i] = (min(nums[i - 1], nums[i + 1]) - 1)
            else:
                if nums[i] >= nums[i - 1]:
                    nums[i] = (nums[i - 1] - 1)

        cnt = sum(backup) - sum(nums)
        nums = backup[:]
        cnt1 = 0
        for i in range(0, len(nums), 2):#保证奇数索引更大，就要让偶数索引变小
            if i != 0 and i != len(nums) - 1:
                if nums[i] >= nums[i - 1] or nums[i] >= nums[i + 1]:
                    nums[i] = (min(nums[i - 1], nums[i + 1]) - 1)
            elif i == 0:
                if nums[i] >= nums[i + 1]:
                    nums[i] = (nums[i + 1] - 1)    
            elif i == len(nums) - 1:
                if nums[i] >= nums[i - 1]:
                    nums[i] = (nums[i - 1] - 1)    
        cnt1 = sum(backup) - sum(nums)
        return min(cnt, cnt1)