如果乙個數列至少有三個元素,並且任意兩個相鄰元素之差相同,則稱該數列為等差數列。
例如,以下數列為等差數列:
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 陣列累加的結果。
classsolution
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 是整...