[leetcode每日一题]2.27

时间:2023-02-27 07:58:38

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

提示

中等

51

相关企业

给你一个整数数组 ​​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​

Solution

简单模拟两种情况:\/\/以及/\/\,只减小奇数位或者偶数位。

class Solution:
def movesToMakeZigzag(self, nums: List[int]) -> int:
n = len(nums)
ans1 = ans2 = 0
if n == 1:
return 0
for i in range(0, n, 2):
if i == 0:
if nums[0] >= nums[1]:
ans1 += nums[0] - nums[1] + 1
elif i == n - 1:
if nums[i] >= nums[i-1]:
ans1 += nums[i] - nums[i-1] + 1
else:
if nums[i] >= nums[i-1] or nums[i] >= nums[i+1]:
ans1 += nums[i] - min(nums[i-1], nums[i+1]) + 1
for i in range(1, n, 2):
if i == n - 1:
if nums[i] >= nums[i-1]:
ans2 += nums[i] - nums[i-1] + 1
else:
if nums[i] >= nums[i-1] or nums[i] >= nums[i+1]:
ans2 += nums[i] - min(nums[i-1], nums[i+1]) + 1
return min(ans1, ans2)

自信一遍过。