以输入样例 [9, 6, 1, 6, 2] 为例,检查奇数位置,可以将第一个元素 9 减小到 5,如果检查偶数位置,则需要将两个 6 减小到 0。
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15
| class Solution: def movesToMakeZigzag(self, nums: List[int]) -> int: odd,even = 0,0 for i in range(len(nums)): if i % 2 == 0: e1 = nums[i] - nums[i-1] + 1 if i > 0 and nums[i] >= nums[i-1] else 0 e2 = nums[i] - nums[i+1] + 1 if i < len(nums) - 1 and nums[i] >= nums[i+1] else 0 even += max(e1,e2) else: o1 = nums[i] - nums[i-1] + 1 if i > 0 and nums[i] >= nums[i-1] else 0 o2 = nums[i] - nums[i+1] + 1 if i < len(nums) - 1 and nums[i] >= nums[i+1] else 0 odd += max(o1,o2) return min(odd,even)
|