当前位置：首页 > web >正文

【leetcode】3356. 零数组变换②

web 2025/8/12 4:35:32

文章目录

- - 1. 暴力求解
  - 2. 差分数组+二分查找
  - - leedcodet评论题解

leetcode：3356. 零数组变换②

给你一个长度为 n 的整数数组 nums 和一个二维数组 queries，其中 queries[i] = [li, ri, vali]。

每个 queries[i] 表示在 nums 上执行以下操作：

将 nums 中 [li, ri] 范围内的每个下标对应元素的值最多减少 vali。
每个下标的减少的数值可以独立选择。

零数组 是指所有元素都等于 0 的数组。

返回 k 可以取到的 最小非负 值，使得在顺序处理前 k 个查询后，nums 变成 零数组。如果不存在这样的 k，则返回 -1。

示例 1：

输入： nums = [2,0,2], queries = [[0,2,1],[0,2,1],[1,1,3]]

输出： 2

解释：

对于 i = 0（l = 0, r = 2, val = 1）：
- 在下标 [0, 1, 2] 处分别减少 [1, 0, 1]。
- 数组将变为 [1, 0, 1]。
对于 i = 1（l = 0, r = 2, val = 1）：
- 在下标 [0, 1, 2] 处分别减少 [1, 0, 1]。
- 数组将变为 [0, 0, 0]，这是一个零数组。因此，k 的最小值为 2。

示例 2：

输入： nums = [4,3,2,1], queries = [[1,3,2],[0,2,1]]

输出： -1

解释：

对于 i = 0（l = 1, r = 3, val = 2）：
- 在下标 [1, 2, 3] 处分别减少 [2, 2, 1]。
- 数组将变为 [4, 1, 0, 0]。
对于 i = 1（l = 0, r = 2, val = 1）：
- 在下标 [0, 1, 2] 处分别减少 [1, 1, 0]。
- 数组将变为 [3, 0, 0, 0]，这不是一个零数组。

1. 暴力求解

暴力求解：倒在621/627

class Solution(object):def minZeroArray(self, nums, queries):""":type nums: List[int]:type queries: List[List[int]]:rtype: int"""# 暴力求解arr = [1] * len(nums)sum0 = 0for num in nums:sum0 += numif sum0 == 0:return 0for epoch, (l, r, val) in enumerate(queries):for i in range(l, r + 1):# 通过数组arr，如果减到0，则之后不用下面的操作if arr[i]:if nums[i] > val:nums[i] -= valelse:nums[i] = 0 arr[i] = 0sum = 0for num in nums:sum += numif sum == 0:return epoch + 1return -1

2. 差分数组+二分查找

class Solution(object):def minZeroArray(self, nums, queries):""":type nums: List[int]:type queries: List[List[int]]:rtype: int"""# 差分数组+二分查找def check(k):diff = [0] * (len(nums) + 1)for l, r, val in queries[:k]:diff[l] += valdiff[r + 1] -= valsum_i = 0for i in range(len(nums)):sum_i += diff[i]if nums[i] > sum_i:return Falsereturn Trueleft = -1right = len(queries) + 1while left + 1 < right:mid = (left + right) // 2# 看中间mid是否满足条件使得零数组if check(mid):right = midelse:left = midreturn right if right <= len(queries) else -1