leetcode.1027 Longest Arithmetic Sequence 最长等差数列

给定一个整数数组 A,返回 A 中最长等差子序列的长度。

回想一下,A 的子序列是列表 A[i_1], A[i_2], ..., A[i_k] 其中 0 <= i_1 < i_2 < ... < i_k <= A.length - 1。并且如果 B[i+1] - B[i]0 <= i < B.length - 1) 的值都相同,那么序列 B 是等差的。

提示:

  1. 2 <= A.length <= 2000
  2. 0 <= A[i] <= 10000

用Go语言写点算法题,熟悉一下这门语言吧~

这道题最好的写法应该是动态规划,而且对使用空间什么的有比较好的优化。

我在这就不讲这些了,能使用Go顺利写出就可以了。

【思路】对每个遍历经过的数字都建立一个diff:len的map,相当于动态规划,但由于用到了map(Go语言里map都是指hashmap),运算时间一般般。时间复杂度还是O(n^2),比暴力求解还是好上不少的。

代码:

func longestArithSeqLength(A []int) int {
    if len(A)<= 2 {
        return len(A)
    }
    size := len(A)
    var dict []map[int]int
    dict = make([]map[int]int,size)//map的slice和map都需要make初始化这一点让我很不习惯
    dict[0] = make(map[int]int)
    for i := 1; i<size;i++{
        dict[i] = make(map[int]int)
        for j:=i-1;j>=0; j--{
            diff := A[i] - A[j]
            if dict[j][diff] + 1>dict[i][diff] {//这里需要判断,否则会产生一个bug
				dict[i][diff] = dict[j][diff] + 1
			}
        }
    }
    res := 2
    for index:=1;index<size;index++{
        for _,v := range dict[index]{
            if v>=res{
                res = v+1
            }
        }
    }
    return res
}

猜你喜欢

转载自www.cnblogs.com/J1ac/p/10859875.html
今日推荐