数据结构
网络
关系数据库管理系统 (RDBMS)
操作系统
Java
MS Excel
iOS
HTML
CSS
Android
Python
C 编程
C++
C#
MongoDB
MySQL
Javascript
PHP
物理学
化学
生物学
数学
英语
经济学
心理学
社会研究
服装研究
法律研究C++中的算术切片 II - 子序列
假设我们有一个数组 A，其中包含 N 个数字。该数组的子序列切片是任何整数序列，例如 (K0, K1, K2, … Kn)，其中 0 <= K0 < K1 < K2 < … < Kn < N。如果序列 A[K0]、A[K1]、… A[Kn] 是算术序列，则 A 的子序列切片 (K0, K1, K2, … Kn) 称为算术切片，这意味着 n >= 2。因此，我们必须返回算术切片的数量。
因此，如果输入类似于 [2,4,6,8,10]，则答案将是 7，因为有 7 个算术切片：[2,4,6]，[2,4,10]，[4,6,8]，[6,8,10]，[2,4,6,8]，[4,6,8,10]，[2,4,6,8,10]。
为了解决这个问题，我们将遵循以下步骤：
- ret := 0
- 定义一个映射 dp 和另一个映射 cnt
- 通过从 A 中获取元素来定义一个集合 s
- n := A 的大小
- 对于初始化 i := 1，当 i < n 时，更新（i 增加 1），执行：
- 对于初始化 j := i - 1，当 j >= 0 时，更新（j 减少 1），执行：
- diff := A[i] - A[j]
- 如果 diff <= -inf 或 diff > inf，则：
- 忽略以下部分，跳到下一个迭代
- 如果 diff 在映射 dp[j] 中，则 temp := dp[j, diff]，否则为 0
- ret := ret + temp
- 如果 (A[i] + diff) 存在于 s 中，则：
- dp[i, diff] := dp[i, diff] + temp + 1
- 对于初始化 j := i - 1，当 j >= 0 时，更新（j 减少 1），执行：
- 返回 ret
让我们看看下面的实现以更好地理解：
示例
#include <bits/stdc++.h>
using namespace std;
typedef long long int lli;
class Solution {
public:
int numberOfArithmeticSlices(vector<int>& A) {
int ret = 0;
unordered_map <lli, unordered_map <lli, lli> > dp, cnt;
unordered_set <int> s (A.begin(), A.end());
int n = A.size();
for(int i = 1; i < n; i++){
for(int j = i - 1; j >= 0; j--){
lli diff = (lli)A[i] - (lli)A[j];
if(diff <= INT_MIN || diff > INT_MAX) continue;
int temp = dp[j].count(diff) ? dp[j][diff] : 0;
ret += temp;
if(s.count(A[i] + diff))dp[i][diff] += temp + 1;
}
}
return ret;
}
};
main(){
Solution ob;
vector<int> v = {2,4,6,8,10};
cout << (ob.numberOfArithmeticSlices(v));
}输入
{2,4,6,8,10}输出
7
更新于：2020年6月1日
164 次浏览
广告