2024.7.24遇到的手撕代码(dp思路,没想出来递推公式,用暴力超时):
. - 力扣(LeetCode)1027.最长等差数列
给你一个整数数组 nums
,返回 nums
中最长等差子序列的长度。
回想一下,nums
的子序列是一个列表 nums[i1], nums[i2], ..., nums[ik]
,且 0 <= i1 < i2 < ... < ik <= nums.length - 1
。并且如果 seq[i+1] - seq[i]
( 0 <= i < seq.length - 1
) 的值都相同,那么序列 seq
是等差的。
示例 1:
输入:nums = [3,6,9,12] 输出:4 解释: 整个数组是公差为 3 的等差数列。
示例 2:
输入:nums = [9,4,7,2,10] 输出:3 解释: 最长的等差子序列是 [4,7,10]。
示例 3:
输入:nums = [20,1,15,3,10,5,8] 输出:4 解释: 最长的等差子序列是 [20,15,10,5]。
提示:
2 <= nums.length <= 1000
0 <= nums[i] <= 500
class Solution:
def longestArithSeqLength(self, nums: List[int]) -> int:
if not nums:
return 0
dp={}
ans=1
for i in range(len(nums)):
for j in range(i):
sub=nums[i]-nums[j]
if (j,sub) in dp:
dp[i,sub]=dp[j,sub]+1
else:
dp[i,sub]=2
ans=max(ans,dp[i,sub])
return ans
同类型的还有另外一道题:
. - 力扣(LeetCode)1218.最长定差子序列
给你一个整数数组 arr
和一个整数 difference
,请你找出并返回 arr
中最长等差子序列的长度,该子序列中相邻元素之间的差等于 difference
。
子序列 是指在不改变其余元素顺序的情况下,通过删除一些元素或不删除任何元素而从 arr
派生出来的序列。
示例 1:
输入: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]。
提示:
1 <= arr.length <= 10^5
-104 <= arr[i], difference <= 10^4
class Solution:
def longestSubsequence(self, arr: List[int], difference: int) -> int:
n=len(arr)
ans=1
dp={}
for num in arr:
if num - difference in dp:
dp[num] = dp[num - difference] + 1
else:
dp[num] = 1
ans = max(ans, dp[num])
return ans