413 等差數列劃分

2022-07-19 08:27:13 字數 1384 閱讀 2078

如果乙個數列至少有三個元素,並且任意兩個相鄰元素之差相同,則稱該數列為等差數列。

例如,以下數列為等差數列:

1, 3, 5, 7, 9

7, 7, 7, 7

3, -1, -5, -9

以下數列不是等差數列。

1, 1, 2, 5, 7

陣列 a 包含 n 個數,且索引從0開始。陣列 a 的乙個子陣列劃分為陣列 (p, q),p 與 q 是整數且滿足 0<=p如果滿足以下條件,則稱子陣列(p, q)為等差陣列:

元素 a[p], a[p + 1], ..., a[q - 1], a[q] 是等差的。並且 p + 1 < q 。

函式要返回陣列 a 中所有為等差陣列的子陣列個數。

示例:

a = [1, 2, 3, 4]

返回: 3, a 中有三個子等差陣列: [1, 2, 3], [2, 3, 4] 以及自身 [1, 2, 3, 4]。

思路:

【0,1,2,3,4】為例

dp[i] 表示以 a[i] 為結尾的等差遞增子區間的個數。

當 a[i] - a[i-1] == a[i-1] - a[i-2],那麼 [a[i-2], a[i-1], a[i]] 構成乙個等差遞增子區間。而且在以 a[i-1] 為結尾的遞增子區間的後面再加上乙個 a[i],一樣可以構成新的遞增子區間。

dp[2] = 1[0, 1, 2]

dp[3] = dp[2] + 1 = 2[0, 1, 2, 3], //

[0, 1, 2] 之後加乙個 3

[1, 2, 3] //

新的遞增子區間

dp[4] = dp[3] + 1 = 3[0, 1, 2, 3, 4], //

[0, 1, 2, 3] 之後加乙個 4

[1, 2, 3, 4], //

[1, 2, 3] 之後加乙個 4

[2, 3, 4] //

新的遞增子區間

綜上,在 a[i] - a[i-1] == a[i-1] - a[i-2] 時,dp[i] = dp[i-1] + 1。

因為遞增子區間不一定以最後乙個元素為結尾,可以是任意乙個元素結尾,因此需要返回 dp 陣列累加的結果。

class

solution

int n =a.length;

int dp = new

int[n];

for (int i = 2; i < n; i++)

}int total = 0;

for (int

cnt : dp)

return

total;

}}

Leetcode 413 等差數列劃分

如果乙個數列至少有三個元素,並且任意兩個相鄰元素之差相同,則稱該數列為等差數列。例如,以下數列為等差數列 1,3,5,7,9 7,7,7,7 3,1,5,9 以下數列不是等差數列。1,1,2,5,7陣列 a 包含 n 個數,且索引從0開始。陣列 a 的乙個子陣列劃分為陣列 p,q p 與 q 是整數...

Leetcode 413 等差數列劃分

如果乙個數列至少有三個元素,並且任意兩個相鄰元素之差相同,則稱該數列為等差數列。例如,以下數列為等差數列 1,3,5,7,9 7,7,7,7 3,1,5,9以下數列不是等差數列。1,1,2,5,7陣列 a 包含 n 個數,且索引從0開始。陣列 a 的乙個子陣列劃分為陣列 p,q p 與 q 是整數且...

Leetcode 413 等差數列劃分

如果乙個數列至少有三個元素,並且任意兩個相鄰元素之差相同,則稱該數列為等差數列。例如,以下數列為等差數列 1,3,5,7,9 7,7,7,7 3,1,5,9 以下數列不是等差數列。1,1,2,5,7 陣列 a 包含 n 個數,且索引從0開始。陣列 a 的乙個子陣列劃分為陣列 p,q p 與 q 是整...