446.等差數列劃分II-子序列
如果一個數列至少有三個元素,並且任意兩個相鄰元素之差相同,則稱該數列為等差數列。
例如,以下數列為等差數列:
1, 3, 5, 7, 9 7, 7, 7, 7 3, -1, -5, -9
以下數列不是等差數列。
1, 1, 2, 5, 7
陣列 A 包含 N 個數,且索引從 0 開始。該陣列子序列將劃分為整數序列 (P0, P1, ..., Pk),P 與 Q 是整數且滿足 0 ≤ P0 < P1 < ... < Pk < N。
如果序列 A[P0],A[P1],...,A[Pk-1],A[Pk] 是等差的,那麼陣列 A 的子序列 (P0,P1,…,PK) 稱為等差序列。值得注意的是,這意味著 k ≥ 2。
函式要返回陣列 A 中所有等差子序列的個數。
輸入包含 N 個整數。每個整數都在 -231 和 231-1 之間,另外 0 ≤ N ≤ 1000。保證輸出小於 231-1。
示例:
輸入:[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]
class Solution {
public:
int numberOfArithmeticSlices(vector<int>& A) {
int res = 0, n = A.size();
vector<unordered_map<int, int>> dp(n);
for (int i = 0; i < n; ++i) {
for (int j = 0; j < i; ++j) {
long delta = (long)A[i] - A[j];
if (delta > INT_MAX || delta < INT_MIN) continue;
int diff = (int)delta;
++dp[i][diff];
if (dp[j].count(diff)) {
res += dp[j][diff];
dp[i][diff] += dp[j][diff];
}
}
}
return res;
}
};