-
#include
#include
using namespace std;
int ArithmeticSlices(vector vec)
{
vector num(vec.size(), 0);
int res = 0;
for (int i = 2; i < vec.size(); i++)
{
if (vec[i] - vec[i - 1] == vec[i - 1] - vec[i - 2])
{
num[i] = num[i - 1] + 1;//当发生这种情况的时候,除了和num[i-1]组合一个外,还多了一个自身。如1 2 3.到4的话,1 2 3 4 肯定是一个,主要是多了一个2 3 4.因为以前2 3是构造不成等差数列的。
123构成1 2 3 。1 2 3 4也是1 2 3、2 3 4、1 2 3 4.5加进去以后可以是1 2 3、 2 3 4、 3 4 5 、1 2 3 4 、2 3 4 5 、1 2 3 4 5
//用DP来做,定义一个一维dp数组,其中dp[i]表示,到i位置为止的Arithmetic Slices的个数,那么我们从第三个数字开始遍历,如果当前数字和之前两个数字构成Arithmetic Slices,那么我们更新dp[i]为dp[i-1]+1
dp[i]=dp[i-1]+1可以理解为,当A[i]可以和A[i-1]以及A[i-2]构成Arithmetic Slices的时候,那么能和A[i-1]构成Arithmetic Slices的元素必定能和A[i]构成算数切片,所以有个dp[i-1],除此之外,当前这个最新元素还可以和它前面的两个元素构成Arithmetic Slices,而它前面的两个元素在dp[i-1]的时候只有两个元素,无法构成Arithmetic Slices,故+1。
}
res += num[i];
}
return res;
}
int ArithmeticSlicesT(vector vec)
{
int res = 0;
int len = 2;
for (int i = 2; i < vec.size(); i++)
{
if (vec[i] - vec[i - 1] == vec[i - 1] - vec[i - 2])
{
len++;
}
else
{
if (len >= 3)
{
res += (len - 1)*(len - 2)*0.5;
}
len = 2;
}}
res += (len - 1)*(len - 2)*0.5;
return res;
}
int main()
{
cout << ArithmeticSlicesT({ 1,2,3,4,5}) << endl;
system(“pause”);
return 0;
}
[LeetCode] Arithmetic Slices 算数切片
最新推荐文章于 2024-10-01 18:52:56 发布
