class Solution {
public:
vector<bool> checkArithmeticSubarrays(vector<int>& nums, vector<int>& l, vector<int>& r) {
}
};
1630. 等差子数组
如果一个数列由至少两个元素组成，且每两个连续元素之间的差值都相同，那么这个序列就是 等差数列 。更正式地，数列 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.lengthm == l.lengthm == r.length2 <= n <= 5001 <= m <= 5000 <= l[i] < r[i] < n-105 <= nums[i] <= 105原站题解
python3 解法, 执行用时: 128 ms, 内存消耗: 15 MB, 提交时间: 2022-11-15 11:36:46
class Solution:
def checkArithmeticSubarrays(self, nums: List[int], L: List[int], R: List[int]) -> List[bool]:
def check(l, r):
if r - l == 1:
return True
A = nums[l : r + 1]
m = min(A)
D = []
for e in A:
if e != m:
D.append(e - m)
if not D:
return True
if len(D) != len(A) - 1:
return False
d = min(D)
vis = [False] * len(D)
for e in D:
if e % d:
return False
tmp = e // d - 1
if tmp < 0 or tmp >= len(D) or vis[tmp]:
return False
else:
vis[tmp] = True
return True
return [check(l, r) for l, r in zip(L, R)]
python3 解法, 执行用时: 100 ms, 内存消耗: 15 MB, 提交时间: 2022-11-15 11:32:20
class Solution:
def checkArithmeticSubarrays(self, nums: List[int], l: List[int], r: List[int]) -> List[bool]:
def is_same_delta(data: List[int]) -> bool:
data.sort()
return all(data[i-1] + data[i+1] == data[i] * 2 for i in range(1, len(data)-1))
n, m = len(nums), len(l)
ans = [True] * m
for i in range(m):
ans[i] = is_same_delta(nums[l[i]:r[i]+1])
return ans