给你一个整数数组 arr 和一个整数 difference,请你找出 arr 中所有相邻元素之间的差等于给定 difference 的等差子序列,并返回其中最长的等差子序列的长度。
示例:
输入:arr = [1,2,3,4], difference = 1
输出:4
解释:最长的等差子序列是 [1,2,3,4]。
示例2
输入:arr = [1,3,5,7], difference = 1
输出:1
解释:最长的等差子序列是任意单个元素。
示例3
输入:arr = [1,5,7,8,5,3,4,2,1], difference = -2
输出:4
解释:最长的等差子序列是 [7,5,3,1]。
// 使用map,是因为可能i值为负数,所以不能直接用数组;
// 思路:
// mmap[i] = 1 表示i是定差子序列的第一个元素,之前的值都不在子序列中
// mmap[i] = mmap[i-difference] + 1 表示i-difference值也在子序列中
// 两种情况可统一为mmap[i] = mmap[i-difference] + 1, mmap的默认值为0.
func longestSubsequence(arr []int, difference int) int {
mmap := make(map[int]int) // mmap[i] = j 表示以i结尾的等差序列长度为j
var res = 0
for i := 0; i < len(arr); i++ {
mmap[arr[i]] = mmap[arr[i]-difference] + 1
if mmap[arr[i]] > res {
res = mmap[arr[i]]
}
}
return res
}
原创声明:本文系作者授权腾讯云开发者社区发表,未经许可,不得转载。
如有侵权,请联系 cloudcommunity@tencent.com 删除。
原创声明:本文系作者授权腾讯云开发者社区发表,未经许可,不得转载。
如有侵权,请联系 cloudcommunity@tencent.com 删除。