动态规划5.0
动态规划 - - - 子序列问题(数组中不连续的一段)
1. 最长递增子序列
题目链接 -> Leetcode -300.最长递增子序列
Leetcode -300.最长递增子序列
题目:给你一个整数数组 nums ,找到其中最长严格递增子序列的长度。
子序列 是由数组派生而来的序列,删除(或不删除)数组中的元素而不改变其余元素的顺序。 例如,[3, 6, 2, 7] 是数组[0, 3, 1, 6, 2, 2, 7] 的子序列。
示例 1: 输入:nums = [10, 9, 2, 5, 3, 7, 101, 18] 输出:4 解释:最长递增子序列是[2, 3, 7, 101],因此长度为 4 。
示例 2: 输入:nums = [0, 1, 0, 3, 2, 3] 输出:4
示例 3: 输入:nums = [7, 7, 7, 7, 7, 7, 7] 输出:1
提示:
- 1 <= nums.length <= 2500
- 10^4 <= nums[i] <= 10^4
思路:
- dp[i] 表示:以 i 位置元素为结尾的「所有子序列」中,最长递增子序列的长度;
- 状态转移方程:对于 dp[i] ,我们可以根据「子序列的构成方式」,进行分类讨论:
- 子序列长度为 1 :只能自己了,此时 dp[i] = 1 ;
- 子序列长度大于 1 : nums[i] 可以跟在前面任何⼀个数后面形成子序列。设前面的某一个数的下标为 j ,其中 0 <= j <= i - 1 。只要 nums[j] < nums[i] , i 位置元素跟在 j 元素后面就可以形成递增序列,长度为 dp[j] 1 ;
因此,我们仅需找到满足要求的最大的 dp[j] 1 即可;
综上, dp[i] = max(dp[j] 1, dp[i]) ,其中 0 <= j <= i - 1 && nums[j] < nums[i];
- 返回值:由于不知道最长递增子序列以谁结尾,因此返回 dp 表里面的「最大值」;
代码如下:
代码语言:javascript复制 class Solution {
public:
int lengthOfLIS(vector<int>& nums)
{
// 动态规划
// dp[i] 表示:以 i 位置元素为结尾的「所有子序列」中,最长递增子序列的长度
vector<int> dp(nums.size(), 1);
int ret = 0;
for (int i = nums.size() - 1; i >= 0; i--)
{
for (int j = i 1; j < nums.size(); j )
{
if (nums[j] > nums[i])
{
dp[i] = max(dp[j] 1, dp[i]);
}
}
ret = max(dp[i], ret);
}
return ret;
}
};
2. 摆动序列
题目链接 -> Leetcode -376.摆动序列
Leetcode -376.摆动序列
题目:如果连续数字之间的差严格地在正数和负数之间交替,则数字序列称为 摆动序列 。第一个差(如果存在的话)可能是正数或负数。仅有一个元素或者含两个不等元素的序列也视作摆动序列。 例如,[1, 7, 4, 9, 2, 5] 是一个 摆动序列 ,因为差值(6, -3, 5, -7, 3) 是正负交替出现的。
相反,[1, 4, 7, 2, 5] 和[1, 7, 4, 5, 5] 不是摆动序列,第一个序列是因为它的前两个差值都是正数,第二个序列是因为它的最后一个差值为零。 子序列 可以通过从原始序列中删除一些(也可以不删除)元素来获得,剩下的元素保持其原始顺序。
给你一个整数数组 nums ,返回 nums 中作为 摆动序列 的 最长子序列的长度 。
示例 1: 输入:nums = [1, 7, 4, 9, 2, 5] 输出:6 解释:整个序列均为摆动序列,各元素之间的差值为(6, -3, 5, -7, 3) 。
示例 2: 输入:nums = [1, 17, 5, 10, 13, 15, 10, 5, 16, 8] 输出:7 解释:这个序列包含几个长度为 7 摆动序列。 其中一个是[1, 17, 10, 13, 10, 16, 8] ,各元素之间的差值为(16, -7, 3, -3, 6, -8) 。
示例 3: 输入:nums = [1, 2, 3, 4, 5, 6, 7, 8, 9] 输出:2
提示:
- 1 <= nums.length <= 1000
- 0 <= nums[i] <= 1000
思路:
- 状态表示:这里我们选择比较常用的方式,以某个位置为结尾,结合题目要求,定义一个状态表示:dp[i] 表示「以 i 位置为结尾的最长摆动序列的长度」。 但是,问题来了,如果状态表示这样定义的话,以 i 位置为结尾的最长摆动序列的长度我们没法从之前的状态推导出来。因为我们不知道前一个最长摆动序列的结尾处是递增的,还是递减的。因此,我们需要状态表示能表示多一点的信息:要能让我们知道这一个最长摆动序列的结尾是递增的还是递减的。
所以我们应该定义两个 dp 表:
- f[i] 表示:以 i 位置元素为结尾的所有的子序列中,最后一个位置呈现「上升趋势」的最长摆动序列的长度;
- g[i] 表示:以 i 位置元素为结尾的所有的子序列中,最后一个位置呈现「下降趋势」的最长摆动序列的长度;
- 状态转移方程:由于子序列的构成比较特殊, i 位置为结尾的子序列,前一个位置可以是 [0, i - 1] 的任意位置,因此设 j 为 [0, i - 1] 区间内的某一个位置。
- 对于 f[i] ,我们可以根据「子序列的构成方式」,进行分类讨论: i. 子序列长度为 1 :只能自己了,此时 f[i] = 1;这种情况不需要处理,我们在初始化的时候将 dp 表初始化为 1 即可 ; ii. 子序列长度大于 1 :因为结尾要呈现上升趋势,因此需要 nums[j] < nums[i] 。在满足这个条件下, j 结尾需要呈现下降状态,最长的摆动序列就是 g[j] 1 。因此我们要找出所有满足条件下的最大的 g[j] 1 。
综上,因为题目需要返回的是最长子序列的长度所以 f[i] 的状态转移方程为 f[i] = max(g[j] 1, f[i]) ,注意使用 g[j] 时需要判断;
- 对于 g[i] ,我们可以根据「子序列的构成方式」,进行分类讨论: i. 子序列长度为 1 :只能自己了,此时 g[i] = 1 ;这种情况不需要处理,我们在初始化的时候将 dp 表初始化为 1 即可 ; ii. 子序列长度大于 1 :因为结尾要呈现下降趋势,因此需要 nums[j] > nums[i] 。在满足这个条件下, j 结尾需要呈现上升状态,因此最长的摆动序列就是 f[j] 1 。因此我们要找出所有满足条件下的最大的 f[j] 1 。
综上, g[i] = max(f[j] 1, g[i]) ,注意使用 f[j] 时需要判断;
- 返回值:应该返回「两个 dp 表里面的最大值」,我们可以在填表的时候,顺便更新⼀个「最大值」;
代码如下:
代码语言:javascript复制 class Solution {
public:
int wiggleMaxLength(vector<int>& nums)
{
int n = nums.size();
// f 表存最后呈现“上升”趋势的最长摆动子序列的长度
// g 表存最后呈现“下降”趋势的最长摆动子序列的长度
vector<int> f(n, 1), g(n, 1);
int ans = 1;
for(int i = 1; i < n; i )
{
for(int j = i - 1; j >= 0; j--)
{
// 如果是上升趋势,f[i] 就是 g 表中的最长摆动子序列的长度 1,因为 g 表是呈下降趋势的;下同
if(nums[i] > nums[j]) f[i] = max(g[j] 1, f[i]);
else if(nums[i] < nums[j]) g[i] = max(f[j] 1, g[i]);
}
ans = max(g[i], max(f[i], ans));
}
return ans;
}
};
3. 最长递增子序列的个数
题目链接 -> Leetcode -673.最长递增子序列的个数
Leetcode -673.最长递增子序列的个数
题目:给定一个未排序的整数数组 nums , 返回最长递增子序列的个数 。
注意 这个数列必须是 严格 递增的。
示例 1: 输入: [1, 3, 5, 4, 7] 输出 : 2 解释 : 有两个最长递增子序列,分别是[1, 3, 4, 7] 和[1, 3, 5, 7]。
示例 2 : 输入 : [2, 2, 2, 2, 2] 输出 : 5 解释 : 最长递增子序列的长度是1,并且存在5个子序列的长度为1,因此输出5。
提示 :
- 1 <= nums.length <= 2000
- -10^6 <= nums[i] <= 10^6
思路:
- 状态表示:先尝试定义一个状态:以 i 为结尾的最长递增子序列的「个数」。那么问题就来了,我们都不知道以 i 为结尾的最长递增子序列的「长度」是多少,我怎么知道最长递增子序列的个数呢?
因此,我们解决这个问题需要两个状态,一个是「长度」,一个是「个数」:
- len[i] 表示:以 i 为结尾的最长递增子序列的长度;
- count[i] 表示:以 i 为结尾的最长递增子序列的个数;
- 状态转移方程:求个数之前,我们得先知道长度,因此先看 len[i] :
- 在求 i 结尾的最长递增序列的长度时,我们已经知道 [0, i - 1] 区间上的 len[j] 信息,用 j 表示 [0, i - 1] 区间上的下标;
- 我们需要的是递增序列,因此 [0, i - 1] 区间上的 nums[j] 只要能和 nums[i] 构成上升序列,那么就可以更新 dp[i] 的值,此时最长长度为 dp[j] 1 ;
- 我们要的是 [0, i - 1] 区间上所有情况下的最大值。
综上所述,对于 len[i] ,我们可以得到状态转移方程为:
- len[i] = max(len[j] 1, len[i]) ,其中 0 <= j < i ,并且 nums[j] < nums[i];
在知道每一个位置结尾的最长递增子序列的长度时,我们来看看能否得到 count[i] :
- 我们此时已经知道 len[i] 的信息,还知道 [0, i - 1] 区间上的 count[j] 信息,用 j 表示 [0, i - 1] 区间上的下标;
- 我们可以再遍历⼀遍 [0, i - 1] 区间上的所有元素,只要能够构成上升序列,并且上升序列的长度等于 dp[i] ,那么我们就把 count[i] 加上 count[j] 的值。这样循环一遍之后,count[i] 存的就是我们想要的值。
综上所述,对于 count[i] ,我们可以得到状态转移方程为:
- count[i] = count[j] ,其中 0 <= j < i ,并且 nums[j] < nums[i] && dp[j] 1 == dp[i] ;
- 返回值:用 retlen 表示最终的最长递增子序列的长度。根据题目要求,我们应该返回所有长度等于 retlen 的子序列的个数;
代码如下:
代码语言:javascript复制 class Solution {
public:
int findNumberOfLIS(vector<int>& nums)
{
int n = nums.size();
// len[i] 表示以 i 位置结尾的最长递增子序列的长度
// count[i] 表示以 i 位置结尾的最长递增子序列的个数
vector<int> len(n, 1), count(n, 1);
int retlen = 1, retcount = 1;
for (int i = 1; i < n; i )
{
for (int j = i - 1; j >= 0; j--)
{
if (nums[i] > nums[j])
{
// 贪心
if (len[j] 1 > len[i])
{
len[i] = len[j] 1;
count[i] = count[j];
}
else if (len[j] 1 == len[i])
{
count[i] = count[j];
}
}
}
// 贪心
if (retlen == len[i])
{
retcount = count[i];
}
else if (len[i] > retlen)
{
retlen = len[i];
retcount = count[i];
}
}
return retcount;
}
};
4. 最长数对链
题目链接 -> Leetcode -646.最长数对链
Leetcode -646.最长数对链
题目:给你一个由 n 个数对组成的数对数组 pairs ,其中 pairs[i] = [lefti, righti] 且 lefti < righti 。 现在,我们定义一种 跟随 关系,当且仅当 b < c 时,数对 p2 = [c, d] 才可以跟在 p1 = [a, b] 后面。我们用这种形式来构造 数对链 。 找出并返回能够形成的 最长数对链的长度 。 你不需要用到所有的数对,你可以以任何顺序选择其中的一些数对来构造。
示例 1: 输入:pairs = [[1, 2], [2, 3], [3, 4]] 输出:2 解释:最长的数对链是[1, 2] ->[3, 4] 。
示例 2: 输入:pairs = [[1, 2], [7, 8], [4, 5]] 输出:3 解释:最长的数对链是[1, 2] ->[4, 5] ->[7, 8] 。
提示:
- n == pairs.length
- 1 <= n <= 1000
- -1000 <= lefti < righti <= 1000
思路:本题思路与第一题 最长递增子序列 的思路类似,我们只需要将数对按照第一个元素排一下序即可处理;
代码如下:
代码语言:javascript复制 class Solution {
public:
int findLongestChain(vector<vector<int>>& pairs)
{
sort(pairs.begin(), pairs.end()); // 对第一个元素排序
int n = pairs.size();
vector<int> dp(n, 1);
int ans = 1;
// dp[i] 表⽰以 i 位置的数对为结尾时,最⻓数对链的⻓度
for(int i = 1; i < n; i )
{
for(int j = i - 1; j >= 0; j--)
{
if(pairs[i][0] > pairs[j][1]) dp[i] = max(dp[j] 1, dp[i]);
}
ans = max(dp[i], ans);
}
return ans;
}
};
5. 最长定差子序列
题目链接 -> Leetcode -1218.最长定差子序列
Leetcode -1218.最长定差子序列
题目:给你一个整数数组 arr 和一个整数 difference,请你找出并返回 arr 中最长等差子序列的长度,该子序列中相邻元素之间的差等于 difference 。
子序列 是指在不改变其余元素顺序的情况下,通过删除一些元素或不删除任何元素而从 arr 派生出来的序列。
示例 1: 输入:arr = [1, 2, 3, 4], difference = 1 输出:4 解释:最长的等差子序列是[1, 2, 3, 4]。
示例 2: 输入:arr = [1, 3, 5, 7], difference = 1 输出:1 解释:最长的等差子序列是任意单个元素。
示例 3: 输入:arr = [1, 5, 7, 8, 5, 3, 4, 2, 1], difference = -2 输出:4 解释:最长的等差子序列是[7, 5, 3, 1]。
提示:
- 1 <= arr.length <= 10^5
- -10^4 <= arr[i], difference <= 10^4
思路:
- 状态表示:
- dp[i] 表示:以 i 位置的元素为结尾所有的子序列中,最长的等差子序列的长度。
- 状态转移方程:对于 dp[i] ,上一个定差子序列的取值定为 arr[i] - difference 。只要找到以上一个数字为结尾的定差子序列长度的 dp[arr[i] - difference] ,然后加上 1 ,就是以 i 为结尾的定差子序列的长度。
因此,这里可以选择使用哈希表做优化。我们可以把「元素, dp[j] 」绑定,放进哈希表中。甚至不用创建 dp 数组,直接在哈希表中做动态规划。
- 返回值:根据「状态表示」,返回整个 dp 表中的最大值;
代码如下:
代码语言:javascript复制 class Solution {
public:
int longestSubsequence(vector<int>& arr, int difference)
{
// dp[i] 表示:以 i 位置的元素为结尾所有的⼦序列中,最长的等差子序列的长度
unordered_map<int, int> hash; // arr[i], dp[i]
hash[arr[0]] = 1;
int n = arr.size();
int ans = 1;
for(int i = 1; i < n; i )
{
hash[arr[i]] = hash[arr[i] - difference] 1;
ans = max(hash[arr[i]], ans);
}
return ans;
}
};
6. 最长的斐波那契子序列的长度
题目链接 -> Leetcode -873.最长的斐波那契子序列的长度
Leetcode -873.最长的斐波那契子序列的长度
题目:如果序列 X_1, X_2, …, X_n 满足下列条件,就说它是 斐波那契式 的:
- n >= 3
- 对于所有 i 2 <= n,都有 X_i X_{ i 1 } = X_{ i 2 }
给定一个严格递增的正整数数组形成序列 arr ,找到 arr 中最长的斐波那契式的子序列的长度。如果一个不存在,返回 0 。 (回想一下,子序列是从原序列 arr 中派生出来的,它从 arr 中删掉任意数量的元素(也可以不删),而不改变其余元素的顺序。例如,[3, 5, 8] 是[3, 4, 5, 6, 7, 8] 的一个子序列)
示例 1: 输入 : arr = [1, 2, 3, 4, 5, 6, 7, 8] 输出 : 5 解释 : 最长的斐波那契式子序列为[1, 2, 3, 5, 8] 。
示例 2: 输入 : arr = [1, 3, 7, 11, 12, 14, 18] 输出 : 3 解释 : 最长的斐波那契式子序列有[1, 11, 12]、[3, 11, 14] 以及[7, 11, 18] 。
提示:
- 3 <= arr.length <= 1000
- 1 <= arr[i] < arr[i 1] <= 10^9
思路:
- 状态表示:这里我们选择比较常用的方式,以某个位置为结尾,结合题目要求,定义一个状态表示:dp[i] 表示:以 i 位置元素为结尾的「所有子序列」中,最长的斐波那契子数列的长度。
但是这里有一个非常致命的问题,那就是我们无法确定 i 结尾的斐波那契序列的样子。这样就会导致我们无法推导状态转移方程,因此我们定义的状态表示需要能够确定一个斐波那契序列。
根据斐波那契数列的特性,我们仅需知道序列里面的最后两个元素,就可以确定这个序列的样子。因此,我们修改我们的状态表示为:
- dp[i][j] 表示:以 i 位置以及 j 位置的元素为结尾的所有的子序列中,最长的斐波那契子序列的长度。规定: i < j.
- 状态转移方程:设 nums[i] = b, nums[j] = c ,那么这个序列的前一个元素就是 a = c - b 。我们根据 a 的情况讨论:
- a 存在,下标为 k ,并且 a < b :此时我们需要以 k 位置以及 i 位置元素为结尾的最长斐波那契子序列的长度,然后再加上 j 位置的元素即可。于是 dp[i][j] = dp[k][i] 1 ;
- a 存在,但是 b < a < c :此时只能两个元素自己了, dp[i][j] = 2 ;
- a 不存在:此时依旧只能两个元素自己了, dp[i][j] = 2 ;
综上,状态转移方程分情况讨论即可。
优化点:我们发现,在状态转移方程中,我们需要确定 a 元素的下标。因此我们可以在 dp 之前,将所有的「元素 下标」绑定在一起,放到哈希表中。
- 返回值:因为不知道最终结果以谁为结尾,因此返回 dp 表中的最大值 ret ;但是 ret 可能小于 3 ,小于 3 的话说明不存在;因此需要判断一下。
代码如下:
代码语言:javascript复制 class Solution {
public:
int lenLongestFibSubseq(vector<int>& arr)
{
int n = arr.size();
unordered_map<int, int> hash; // arr[i] - i ,将数组中的元素与下标绑定放入哈希表中
for (int i = 0; i < n; i ) hash[arr[i]] = i;
vector<vector<int>> dp(n, vector<int>(n, 2));
int ans = 2;
// dp[j][i] 表示以 j 和 i 位置的元素为结尾的所有子序列中,最长的斐波那契子序列的长度
for (int i = 2; i < n; i ) // 固定 i 在后
{
for (int j = i - 1; j >= 1; j--) //固定 j 在前
{
int tp = arr[i] - arr[j];
if (tp < arr[j] && hash.count(tp)) dp[j][i] = dp[hash[tp]][j] 1;
ans = max(dp[j][i], ans);
}
hash[arr[i]] = i;
}
return ans < 3 ? 0 : ans;
}
};
7. 最长等差数列
题目链接 -> Leetcode -1027.最长等差数列
Leetcode -1027.最长等差数列
题目:给你一个整数数组 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
思路:本题思路与上题思路类似,只是在推前一个元素的方法不一样,其中本题:设 nums[i] = b, nums[j] = c ,那么这个序列的前一个元素就是 a = 2 * b - c ;接下来我们和上题根据 a 的情况讨论,这里就不再做讨论。
代码如下:
代码语言:javascript复制 class Solution {
public:
int longestArithSeqLength(vector<int>& nums)
{
int n = nums.size();
unordered_map<int, int> hash; // nums[i], i
//for(int i = 0; i < n; i ) hash[nums[i]] = i;
hash[nums[0]] = 0;
int ans = 2;
vector<vector<int>> dp(n, vector<int>(n, 2));
// dp[i][j] 表示:以 i 位置以及 j 位置的元素为结尾的所有的⼦序列中,最长的等差序列的⻓度。规定 i < j
for(int i = 1; i < n; i ) // 固定倒数第二个数
{
for(int j = i 1; j < n; j ) // 固定倒数第一个数
{
// 求出以 i 和 j 位置结尾的子序列中,以它们为等差数列的前一个数
int tp = 2 * nums[i] - nums[j];
if(hash.count(tp))
{
dp[i][j] = dp[hash[tp]][i] 1;
}
ans = max(ans, dp[i][j]);
}
// 一边做 dp 一边更新hash数组;以便多个 tp 元素重复时,hash[tp]找到的是离 i 最近的下标
hash[nums[i]] = i;
}
return ans;
}
};
8. 等差数列划分Ⅱ - 子序列
题目链接 -> 等差数列划分Ⅱ - 子序列
Leetcode -446.等差数列划分Ⅱ - 子序列
题目:给你一个整数数组 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 整数。
示例 1: 输入:nums = [2, 4, 6, 8, 10] 输出:7 解释:所有的等差子序列为: [2, 4, 6] [4, 6, 8] [6, 8, 10] [2, 4, 6, 8] [4, 6, 8, 10] [2, 4, 6, 8, 10] [2, 6, 10]
示例 2: 输入:nums = [7, 7, 7, 7, 7] 输出:16 解释:数组中的任意子序列都是等差子序列。
提示:
- 1 <= nums.length <= 1000
- -2^31 <= nums[i] <= 2^31 - 1
思路:
本题与最长等差数列的区别是,最长等差数列求的是最长等差数列的长度,本题求的是等差数列的数目;所以思路我们也是类似的:
- 状态表示:dp[i][j] 表示:以 i 位置以及 j 位置的元素为结尾的所有的子序列中,等差子序列的个数。规定⼀下 i < j.
- 状态转移方程:设 nums[i] = b, nums[j] = c ,那么这个序列的前一个元素就是 a = 2 * b - c 。我们根据 a 的情况讨论:
- a 存在,下标为 k ,并且 a < b :此时我们知道以 k 元素以及 i 元素结尾的等差序列的个数 dp[k][i] ,在这些子序列的后面加上 j 位置的元素依旧是等差序列。但是这里会多出来一个以 k, i, j 位置的元素组成的新的等差序列,因此 dp[i][j] = dp[k][i] 1 ;
- 因为 a 可能有很多个,我们需要全部累加起来。
综上, dp[i][j] = dp[k][i] 1.
- 返回值:我们要统计所有的等差子序列,因此返回 dp 表中所有元素的和。
代码如下:
代码语言:javascript复制 class Solution {
public:
int numberOfArithmeticSlices(vector<int>& nums)
{
int n = nums.size();
// 将数组的元素和下标数组绑定在一起放入哈希表中,因为相同的元素会有不同的下标
unordered_map<long long, vector<int>> hash;
for(int i = 0; i < n; i ) hash[nums[i]].push_back(i);
vector<vector<int>> dp(n, vector<int>(n));
int count = 0;
for(int i = 1; i < n; i ) // 枚举倒数第二个数
{
for(int j = i 1; j < n; j ) // 枚举倒数第一个数
{
long long tp = (long long)2 * nums[i] - nums[j];
if(hash.count(tp))
{
// 遍历 tp 元素的下标数组,tp 的下标需要小于 i
for(auto& k : hash[tp])
{
// 需要 = ,是因为要确定以 i、j 为结尾的元素前面的等差数列个数,就要用以 k、i 为结尾的元素前面的等差数列个数再加上1;因为以 k、i 为结尾的元素前面的等差数列再加上 j 的元素,也能构成数量相同的等差数列个数,而且多了 k、i、j 这一组等差数列,所以要多加1
if(k < i) dp[i][j] = dp[k][i] 1;
}
}
count = dp[i][j];
}
}
return count;
}
};