｜判斷能否形成等差數(shù)列

給你一個(gè)數(shù)字?jǐn)?shù)組 arr 。如果一個(gè)數(shù)列中，任意相鄰兩項(xiàng)的差總等于同一個(gè)常數(shù)，那么這個(gè)數(shù)列就稱為 等差數(shù)列 。如果可以重新排列數(shù)組形成等差數(shù)列，請(qǐng)返回 true ；否則，返回 false 。示例 1：輸入：arr = [3,5,1] 輸出：true 解釋：對(duì)數(shù)組重新排序得到 [1,3,5] 或者 [5,3,1] ，任意相鄰兩項(xiàng)的差分別為 2 或 -2 ， 可以形成等差數(shù)列。 示例 2：輸入：arr = [1,2,4] 輸出：false 解釋：無法通過重新排序得到等差數(shù)列。提示：2 <= arr.length <= 1000 -10^6 <= arr[i] <= 10^6

題解

class Solution:"""解題思路: 1.先排序，再判斷是否為等差數(shù)列1.直接使用sort 函數(shù)"""def canMakeArithmeticProgression(self, arr: List[int]) -> bool:length = len(arr)if length <= 2:return Truearr.sort()return all([arr[i] - arr[i-1] == arr[-1] - arr[-2] for i in range(length-1, 0, -1)])class Solution:"""解題思路: 1.先排序，再判斷是否為等差數(shù)列2.手動(dòng)實(shí)現(xiàn)快排"""def canMakeArithmeticProgression(self, arr: List[int]) -> bool:length = len(arr)if length <= 2:return Trueself.quick_sort(arr, 0, length-1)return all([arr[i] - arr[i-1] == arr[-1] - arr[-2] for i in range(length-1, 0, -1)])def quick_sort(self, arr, first, last):# 初始化p, q指針if first >= last:returnp, q = first, lastwhile q > p:if arr[p] < arr[first]:p += 1elif arr[q] >= arr[first]:q -= 1else:arr[p], arr[q] = arr[q], arr[p]arr[q] = arr[first]# 對(duì)比左側(cè)元素self.quick_sort(arr, first, q-1)self.quick_sort(arr, q+1, last)

總結(jié)

以上是生活随笔為你收集整理的python判断能否形成等差数列的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。