力扣题解( 等差数列划分 II - 子序列)

446. 等差数列划分 II - 子序列

给你一个整数数组 nums ,返回 nums 中所有 等差子序列 的数目。

如果一个序列中 至少有三个元素 ,并且任意两个相邻元素之差相同,则称该序列为等差序列。

  • 例如,[1, 3, 5, 7, 9][7, 7, 7, 7][3, -1, -5, -9] 都是等差序列。
  • 再例如,[1, 1, 2, 5, 7] 不是等差序列。

数组中的子序列是从数组中删除一些元素(也可能不删除)得到的一个序列。

  • 例如,[2,5,10][1,2,1,2 ,4,1,5 ,10] 的一个子序列。

题目数据保证答案是一个 32-bit 整数。

思路:

首先,本道题要求的是所有可能的子序列的数目,要注意此时子序列可以重复出现,因为是友不同位置的同一元素构成的,比如:{5,5,10,15},可以构成两个子序列{5,10,15},{5,10,15},不同点在于用的5来自不同下标。

再者,对于所求的数量与前面所求数量有所关联的题,往往会出现dpi=dpj+1,这里的加一是指会额外多出一种可能,dpj表示在原本所有j位置构成的字串加上i位置仍是字串,因此数量不变。

对于本题,由于在找前面符合条件的字串时,涉及到了前面字串的具体构成(因为构成字串最少需要三个元素),因此需要用多维数组表示。dpij表示i位置数据在前,j位置数据在后,在加上i之前的元素构成子序列,符合条件的数值是2*numi-numj(因为是等差数列)。然后,由于可以出现重复的子序列,因此对于i位置之前所有符合数值的下标都需要进行加上。

复制代码
class Solution {
public:
    int numberOfArithmeticSlices(vector<int>& nums) {
     int n=nums.size();
     vector<vector<int>>dp(n,vector<int>(n));
     unordered_map<long long,vector<int>>hash;
     int sum=0;
     for(int i=0;i<n;i++)
     {
        for(int j=i+1;j<n;j++)
        {  
            long long a=(long long)2*nums[i]-nums[j];
           if(hash.count(a))
           {
              for(auto k:hash[a])
              if(k<i)
              {
                dp[i][j]+=dp[k][i]+1;
            
              }
           }
           sum+=dp[i][j];
    
        }
        hash[nums[i]].push_back(i);
        
     }

     return sum;



    }
};
相关推荐
mCell4 小时前
Lua 编程入门:从基础语法到元表
javascript·算法·lua
wuyk5556 小时前
98.C语言易混难点:字符数组与字符串指针的底层差异
c语言·开发语言·c++·stm32·嵌入式硬件·算法
峥无6 小时前
从0到1手撕红黑树:封装实现 my_map 与 my_set(SGI-STL 源码级深度解析)
开发语言·c++·笔记·算法·stl
不会代码的小猴7 小时前
3. 控件学习1
开发语言·c++·笔记·qt·算法
疯狂打码的少年8 小时前
【数据结构】图的存储结构:邻接矩阵与邻接表
数据结构·笔记
203号居民8 小时前
LeetCode hot 100 —41. 缺失的第一个正数
数据结构·算法·leetcode
专注仿真11 小时前
问答大模型技术方案算法实现-熵权法融合算法 + 交叉编码器重排算法
人工智能·python·算法·语言模型·问答大模型关键算法·熵权融合算法·交叉编码器重排算法
地平线开发者12 小时前
量化为什么会有损失:从量化误差到精度下降
算法
手写码匠12 小时前
华为云Flexus+DeepSeek征文|Dify 多 Agent 会话管理与上下文工程实战:让智能体“记住该记住的,忘掉该忘掉的“
人工智能·深度学习·算法·aigc