您现在的位置是:首页 >学无止境 >LeetCode_动态规划_中等_1027.最长等差数列网站首页学无止境

LeetCode_动态规划_中等_1027.最长等差数列

代码星辰 2023-06-04 08:00:02
简介LeetCode_动态规划_中等_1027.最长等差数列

1.题目

给你一个整数数组 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

来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/longest-arithmetic-subsequence

2.思路

(1)动态规划
思路参考本题官方题解

① 设 minNummaxNum 分别为数组 nums 中的最大值和最小值,那么子序列的公差 d 的取值范围为 [minNum - maxNum, maxNum - minNum]

② 枚举所有可能的公差 d,在每次枚举的过程中定义以下 dp 数组并进行初始化,dp[i] 表示以 nums[i] 结尾且公差为 d 的最长子序列长度

 int[] dp = new int[maxNum + 1];
 //因为以 nums[i] 结尾且公差为 d 的最长子序列不一定存在,因此先将其值初始化为 -1,方便后续进行判断
 Arrays.fill(dp, -1);

③ 对于每一个 d,遍历数组 nums 中的每一个元素 num:

  • 如果 prev = num - d 未超出数组值范围的边界,并且 dp[prev] != -1,则说明以 nums[prev] 结尾且公差为 d 的最长子序列存在,此时则需要更新 dp[num] 与 res:
dp[num] = Math.max(dp[num], dp[prev] + 1);
res = Math.max(res, dp[num]);
  • 否则,将 dp[num] 更新为 Math.max(dp[num], 1)

④ 枚举完 d 后,直接返回 res 即可。

3.代码实现(Java)

//思路1————
class Solution {
    public int longestArithSeqLength(int[] nums) {
        int n = nums.length;
        int minNum = Arrays.stream(nums).min().getAsInt();
        int maxNum = Arrays.stream(nums).max().getAsInt();
        int diff = maxNum - minNum;
        int res = 2;
        //枚举所有可能的公差
        for (int d = -diff; d <= diff; d++) {
            // dp[i] 表示以 nums[i] 结尾且公差为 d 的最长子序列长度
            int[] dp = new int[maxNum + 1];
            //因为以 nums[i] 结尾且公差为 d 的最长子序列不一定存在,因此先将其值初始化为 -1,方便后续进行判断
            Arrays.fill(dp, -1);
            for (int num : nums) {
                int prev = num - d;
                /*
                    prev >= minNum && prev <= maxNum: 判断边界
                    dp[prev] != -1 说明以 nums[prev] 结尾且公差为 d 的最长子序列存在
                */
                if (prev >= minNum && prev <= maxNum && dp[prev] != -1) {
                    //更新 dp[num] 与 res
                    dp[num] = Math.max(dp[num], dp[prev] + 1);
                    res = Math.max(res, dp[num]);
                }
                dp[num] = Math.max(dp[num], 1);
            }
        }
        return res;
    }
}
风语者!平时喜欢研究各种技术,目前在从事后端开发工作,热爱生活、热爱工作。