网络异常,图片无法展示
|
题目
如果一个数列由至少两个元素组成,且每两个连续元素之间的差值都相同,那么这个序列就是 等差数列 。更正式地,数列 s 是等差数列,只需要满足:对于每个有效的 i , s[i+1] - s[i] == s[1] - s[0] 都成立。
例如,下面这些都是 等差数列 :
1, 3, 5, 7, 9 7, 7, 7, 7 3, -1, -5, -9 下面的数列 不是等差数列 :
1, 1, 2, 5, 7 给你一个由 n 个整数组成的数组 nums,和两个由 m 个整数组成的数组 l 和 r,后两个数组表示 m 组范围查询,其中第 i 个查询对应范围 [l[i], r[i]] 。所有数组的下标都是 从 0 开始 的。
返回 boolean 元素构成的答案列表 answer 。如果子数组 nums[l[i]], nums[l[i]+1], ... , nums[r[i]] 可以 重新排列 形成 等差数列 ,answer[i] 的值就是 true;否则answer[i] 的值就是 false 。
示例 1: 输入:nums = [4,6,5,9,3,7], l = [0,0,2], r = [2,3,5] 输出:[true,false,true] 解释: 第 0 个查询,对应子数组 [4,6,5] 。可以重新排列为等差数列 [6,5,4] 。 第 1 个查询,对应子数组 [4,6,5,9] 。无法重新排列形成等差数列。 第 2 个查询,对应子数组 [5,9,3,7] 。可以重新排列为等差数列 [3,5,7,9] 。 示例 2: 输入:nums = [-12,-9,-3,-12,-6,15,20,-25,-20,-15,-10], l = [0,1,6,4,8,7], r = [4,4,9,7,9,10] 输出:[false,true,false,false,true,true]
提示:
n == nums.length m == l.length m == r.length 2 <= n <= 500 1 <= m <= 500 0 <= l[i] < r[i] < n -105 <= nums[i] <= 105
解题思路
class Solution: def checkArithmeticSubarrays(self, nums: List[int], l: List[int], r: List[int]) -> List[bool]: # 暴力法 # fanweiList = [(l[i], r[i]) for i in range(len(l))] fanweiList = zip(l,r) resList = [] from collections import Counter for x,y in fanweiList: #抽取指定子数组并排序 newNums = sorted(nums[x:y+1]) # print(newNums) #用zip进行错位压缩,然后xy互减,形成等差数组,再通过Counter得出是否只有一个值,判断是否为等差 zipList = list(zip(newNums[:],newNums[1:])) dengchaList = [y-x for x, y in zipList] resCount = Counter(dengchaList) if len(resCount) == 1: resList.append(True) else: resList.append(False) return resList if __name__ == '__main__': nums = [4,6,5,9,3,7] l = [0,0,2] r = [2,3,5] ret = Solution().checkArithmeticSubarrays(nums, l, r) print(ret)