1. 程式人生 > >446.等差數列劃分II-子序列

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