LeetCode 1630. 等差子数组

原创

freesan44

修改于 2021-09-06 11:04:49

2670

修改于 2021-09-06 11:04:49

文章被收录于专栏：freesan44

题目

如果一个数列由至少两个元素组成，且每两个连续元素之间的差值都相同，那么这个序列就是等差数列。更正式地，数列 s 是等差数列，只需要满足：对于每个有效的 i ， si+1 - si == s1 - s0 都成立。

例如，下面这些都是等差数列：

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 个查询对应范围 [li, ri] 。所有数组的下标都是从 0 开始的。

返回 boolean 元素构成的答案列表 answer 。如果子数组 nums[li], nums[li+1], ... , nums[ri] 可以重新排列形成等差数列，answeri 的值就是 true；否则answeri 的值就是 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 <= li < ri < n

-105 <= numsi <= 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)

原创声明：本文系作者授权腾讯云开发者社区发表，未经许可，不得转载。

如有侵权，请联系 cloudcommunity@tencent.com 删除。

python