力扣1027. 最长等差数列

动态规划

  • 思路:
    • 可以参考力扣1218. 最长定差子序列
    • 目前不清楚公差,可以将序列最大最小值找到,公差的范围是 [-(max - min), (max - min)],按公差递增迭代遍历求出最长等差数列;
cpp 复制代码
class Solution {
public:
    int longestArithSeqLength(vector<int>& nums) {
        auto [minit, maxit] = std::minmax_element(nums.begin(), nums.end());
        int diff = *maxit - *minit;
        int ans = 0;
        for (int d = -diff; d <= diff; ++d) {
            std::unordered_map<int, int> dp;
            for (int v : nums) {
                dp[v] = dp[v - d] + 1;
                ans = std::max(ans, dp[v]);
            }
        }

        return ans;
    }
};
  • 时间复杂度比较高,应该是哈希表频繁插入导致,将 dp 数据结构换成数组,数组下标最大值为元素最大值 + 1;
cpp 复制代码
class Solution {
public:
    int longestArithSeqLength(vector<int>& nums) {
        auto [minit, maxit] = std::minmax_element(nums.begin(), nums.end());
        int diff = *maxit - *minit;
        int ans = 1;
        for (int d = -diff; d <= diff; ++d) {
            std::vector<int> dp(*maxit + 1, -1);
            for (int v : nums) {
                int prev = v - d;
                // ensure prev is in nums and has exist(or v is the first item)
                if (prev >= *minit && prev <= *maxit && dp[prev] != -1) {
                    dp[v] = std::max(dp[v], dp[prev] + 1);
                    ans = std::max(ans, dp[v]);
                }
                dp[v] = std::max(dp[v], 1);
            }
        }

        return ans;
    }
};

相关推荐
im_AMBER1 小时前
数据结构 06 线性结构
数据结构·学习·算法
earthzhang20213 小时前
【1028】字符菱形
c语言·开发语言·数据结构·c++·算法·青少年编程
papership3 小时前
【入门级-算法-3、基础算法:二分法】
数据结构·算法
通信小呆呆3 小时前
收发分离多基地雷达椭圆联合定位:原理、算法与误差分析
算法·目标检测·信息与通信·信号处理
丁浩6667 小时前
Python机器学习---2.算法:逻辑回归
python·算法·机器学习
伏小白白白8 小时前
【论文精度-2】求解车辆路径问题的神经组合优化算法:综合展望(Yubin Xiao,2025)
人工智能·算法·机器学习
无敌最俊朗@9 小时前
数组-力扣hot56-合并区间
数据结构·算法·leetcode
囚生CY9 小时前
【速写】优化的深度与广度(Adam & Moun)
人工智能·python·算法
码农多耕地呗9 小时前
力扣94.二叉树的中序遍历(递归and迭代法)(java)
数据结构·算法·leetcode
微笑尅乐9 小时前
BFS 与 DFS——力扣102.二叉树的层序遍历
leetcode·深度优先·宽度优先