力扣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;
    }
};

相关推荐
好家伙VCC42 分钟前
数学建模模型 全网最全 数学建模常见算法汇总 含代码分析讲解
大数据·嵌入式硬件·算法·数学建模
liulilittle2 小时前
IP校验和算法:从网络协议到SIMD深度优化
网络·c++·网络协议·tcp/ip·算法·ip·通信
bkspiderx4 小时前
C++经典的数据结构与算法之经典算法思想:贪心算法(Greedy)
数据结构·c++·算法·贪心算法
中华小当家呐5 小时前
算法之常见八大排序
数据结构·算法·排序算法
沐怡旸5 小时前
【算法--链表】114.二叉树展开为链表--通俗讲解
算法·面试
一只懒洋洋6 小时前
K-meas 聚类、KNN算法、决策树、随机森林
算法·决策树·聚类
方案开发PCBA抄板芯片解密7 小时前
什么是算法:高效解决问题的逻辑框架
算法
songx_997 小时前
leetcode9(跳跃游戏)
数据结构·算法·游戏
小白狮ww8 小时前
RStudio 教程:以抑郁量表测评数据分析为例
人工智能·算法·机器学习