LeetCode-day41-3152. 特殊数组 II
- 题目描述
- 示例
- 示例1:
- 示例2:
- 思路
- 代码
题目描述
如果数组的每一对相邻元素都是两个奇偶性不同的数字,则该数组被认为是一个 特殊数组 。
你有一个整数数组 nums 和一个二维整数矩阵 queries,对于 queries[i] = [fromi, toi],请你帮助你检查 子数组 nums[fromi…toi] 是不是一个 特殊数组 。
返回布尔数组 answer,如果 nums[fromi…toi] 是特殊数组,则 answer[i] 为 true ,否则,answer[i] 为 false 。
示例
示例1:
输入:nums = [3,4,1,2,6], queries = [[0,4]]
输出:[false]
解释:
子数组是 [3,4,1,2,6]。2 和 6 都是偶数。
示例2:
输入:nums = [4,3,1,6], queries = [[0,2],[2,3]]
输出:[false,true]
解释:
子数组是 [4,3,1]。3 和 1 都是奇数。因此这个查询的答案是 false。
子数组是 [1,6]。只有一对:(1,6),且包含了奇偶性不同的数字。因此这个查询的答案是 true。
思路
记录每个位置的最左特殊数组位置
我们可以定义一个数组 d 来记录每个位置的最左特殊数组位置,初始时 d[i]=i。然后我们从左到右遍历数组 nums,如果 nums[i] 和 nums[i−1] 的奇偶性不同,那么 d[i]=d[i−1]。
最后我们遍历每个查询,判断 d[to]<=from 是否成立即可。
代码
class Solution:def isArraySpecial(self, nums: List[int], queries: List[List[int]]) -> List[bool]:n = len(nums)d = list(range(n))for i in range(1,n):if nums[i] % 2 != nums[i - 1] % 2:d[i] = d[i - 1]return [d[t] <= f for f,t in queries]