1. 题目描述
给你一个整数数组nums每次 操作会从中选择一个元素并将该元素的值减少1。
如果符合下列情况之一，则数组A就是锯齿数组：
每个偶数索引对应的元素都大于相邻的元素，即A[0] > A[1] < A[2] > A[3] < A[4] > ...
或者，每个奇数索引对应的元素都大于相邻的元素，即A[0] < A[1] > A[2] < A[3] > A[4] < ...
返回将数组nums转换为锯齿数组所需的最小操作次数。
2. 题目示例
2.1 示例1
输入：nums = [1,2,3]
输出：2
解释：我们可以把2递减到0，或把3递减到1。
2.2 示例2
输入：nums = [9,6,1,6,2]
输出：4
3. 解题代码
class Solution:
def movesToMakeZigzag(self, nums: List[int]) -> int:
n = len(nums)
ans1, ans2 = 0, 0
for i in range(n):
# 奇数位置
if i % 2 == 0:
d1 = nums[i] - nums[i - 1] + 1 if i > 0 and nums[i] >= nums[i - 1] else 0
d2 = nums[i] - nums[i + 1] + 1 if i < n - 1 and nums[i] >= nums[i + 1] else 0
ans1 += max(d1, d2)
# 偶数位置
else:
d1 = nums[i] - nums[i - 1] + 1 if nums[i] >= nums[i - 1] else 0
d2 = nums[i] - nums[i + 1] + 1 if i < n - 1 and nums[i] >= nums[i + 1] else 0
ans2 += max(d1, d2)
return min(ans1, ans2)
另一种写法代码
class Solution:
def movesToMakeZigzag(self, nums: List[int]) -> int:
if len(nums) == 1:
return 0
cur1,cur2 = 0,0
# 计算第一种情况
for i in range(1,len(nums),2):
if (i+1) >= len(nums):
if nums[i-1] <= nums[i]:
cur1 += nums[i] - nums[i-1] + 1
else:
curmin = min(nums[i-1],nums[i+1])
if curmin <= nums[i]:
cur1 += nums[i] - curmin + 1
# 计算第二种情况
for i in range(0,len(nums),2):
if i == 0:
if nums[i] >= nums[i+1]:
cur2 += nums[i] - nums[i+1] + 1
elif i == len(nums)-1:
if nums[i] >= nums[i-1]:
cur2 += nums[i] - nums[i-1] + 1
else:
curmin = min(nums[i-1],nums[i+1])
if curmin <= nums[i]:
cur2 += nums[i] - curmin + 1
# 返回最小的值
return min(cur1,cur2)
