【leetcode】3355. 零数组变换Ⅰ
文章目录
- 1. 暴力求解
- 2. 暴力求解(数组)
- 3. 差分数组
- 解释
给定一个长度为 n 的整数数组 nums 和一个二维数组 queries,其中 queries[i] = [li, ri]。
对于每个查询 queries[i]:
在 nums 的下标范围 [li, ri] 内选择一个下标 子集。
将选中的每个下标对应的元素值减 1。
零数组 是指所有元素都等于 0 的数组。
如果在按顺序处理所有查询后,可以将 nums 转换为 零数组 ,则返回 true,否则返回 false。
示例 1:
输入: nums = [1,0,1], queries = [[0,2]]
输出: true
解释:
对于 i = 0:
选择下标子集 [0, 2] 并将这些下标处的值减 1。
数组将变为 [0, 0, 0],这是一个零数组。
示例 2:
输入: nums = [4,3,2,1], queries = [[1,3],[0,2]]
输出: false
解释:
对于 i = 0:
选择下标子集 [1, 2, 3] 并将这些下标处的值减 1。
数组将变为 [4, 2, 1, 0]。
对于 i = 1:
选择下标子集 [0, 1, 2] 并将这些下标处的值减 1。
数组将变为 [3, 1, 0, 0],这不是一个零数组。
1. 暴力求解
倒在662/668
class Solution(object):def isZeroArray(self, nums, queries):""":type nums: List[int]:type queries: List[List[int]]:rtype: bool"""n = len(nums)m = len(queries)for i in range(m):l = queries[i][0]r = queries[i][1]for j in range(l, r + 1):if nums[j] >= 1:nums[j] -= 1for num in nums:if num != 0:return Falsereturn True
2. 暴力求解(数组)
倒在662/668
class Solution(object):def isZeroArray(self, nums, queries):""":type nums: List[int]:type queries: List[List[int]]:rtype: bool"""# 传统方法:构建arr,在queries所有区间都+1,若nums[i]>arr[i]则不能arr = [0] * len(nums)for l, r in queries:for i in range(l, r + 1):arr[i] += 1for i in range(len(nums)):if nums[i] > arr[i]:return Falsereturn True
3. 差分数组
class Solution(object):def isZeroArray(self, nums, queries):""":type nums: List[int]:type queries: List[List[int]]:rtype: bool"""# 传统方法:构建arr,在queries所有区间都+1,若nums[i]>arr[i]则不能,倒在662/668# 使用差分数组:l作为进水口,r+1作为出水口(后面累加前面)diff = [0] * (len(nums)+1)for l, r in queries:diff[l] += 1diff[r + 1] -= 1sum = 0for i in range(len(nums)):sum += diff[i]if nums[i] > sum:return Falsereturn True
解释
来自于leetcode评论