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

相关推荐
烦躁的大鼻嘎14 分钟前
模拟算法实例讲解:从理论到实践的编程之旅
数据结构·c++·算法·leetcode
C++忠实粉丝31 分钟前
计算机网络socket编程(4)_TCP socket API 详解
网络·数据结构·c++·网络协议·tcp/ip·计算机网络·算法
祁思妙想1 小时前
10.《滑动窗口篇》---②长度最小的子数组(中等)
leetcode·哈希算法
用户37791362947551 小时前
【循环神经网络】只会Python,也能让AI写出周杰伦风格的歌词
人工智能·算法
福大大架构师每日一题1 小时前
文心一言 VS 讯飞星火 VS chatgpt (396)-- 算法导论25.2 1题
算法·文心一言
EterNity_TiMe_1 小时前
【论文复现】(CLIP)文本也能和图像配对
python·学习·算法·性能优化·数据分析·clip
ifanatic1 小时前
[面试]-golang基础面试题总结
面试·职场和发展·golang
机器学习之心1 小时前
一区北方苍鹰算法优化+创新改进Transformer!NGO-Transformer-LSTM多变量回归预测
算法·lstm·transformer·北方苍鹰算法优化·多变量回归预测·ngo-transformer
yyt_cdeyyds2 小时前
FIFO和LRU算法实现操作系统中主存管理
算法
alphaTao2 小时前
LeetCode 每日一题 2024/11/18-2024/11/24
算法·leetcode