LeetCode:判断能否形成等差数列


题目

给你一个数字数组 arr

如果一个数列中,任意相邻两项的差总等于同一个常数,那么这个数列就称为 等差数列

如果可以重新排列数组形成等差数列,请返回 true ;否则,返回 false

示例 1:

1
2
3
输入:arr = [3,5,1]
输出:true
解释:对数组重新排序得到 [1,3,5] 或者 [5,3,1] ,任意相邻两项的差分别为 2 或 -2 ,可以形成等差数列。

示例 2:

1
2
3
输入:arr = [1,2,4]
输出:false
解释:无法通过重新排序得到等差数列。

提示:

  • 2 <= arr.length <= 1000
  • 10^6 <= arr[i] <= 10^6

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/can-make-arithmetic-progression-from-sequence

代码

Go

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
package main

import (
"fmt"
"sort"
)

func canMakeArithmeticProgression(arr []int) bool {
sort.Ints(arr)
step := arr[1] - arr[0]

for i := 1; i < len(arr)-1; i++ {
if arr[i+1]-arr[i] != step {
return false
}
}

return true
}

func main() {
fmt.Println(canMakeArithmeticProgression([]int{7, 1, 3, 5}))
}

评论