以输入样例 [9, 6, 1, 6, 2] 为例,检查奇数位置,可以将第一个元素 9 减小到 5,如果检查偶数位置,则需要将两个 6 减小到 0。
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)