446. 等差数列划分 II - 子序列
[7,7,7,7,7]
思考它为什么会有16种情况。
[7,7,7,7,7]
贪心算法第一次尝试(失败的题解)
class Solution {
public int numberOfArithmeticSlices(int[] nums) {
int len = nums.length;
int res =0;
for(int i = 0;i<len-2;i++){
int pointer = nums[i];
for(int j=i+1;j<len-1;j++){
//第二个值为temp,如果有下一个满足条件的数列项,temp = temp+cut
int temp = nums[j];
int cut = pointer - temp; //[1,2,3] cut = -1 next=temp-cut [3,2,1] cut = 1,next = temp-cut
int next = temp - cut;
for(int z= j+1;z<len;z++){
if(nums[z]==next){
int third = next;//找到了第三项
res++;
int counter = 1;
for(int m= z+1;m<len;m++){
//找第三项之外的项,这里明显出现了思维漏洞,它也可以隔离一项,但是直接遍历忽略了这种情况。
if(third+(cut*counter)==nums[m]){
res++;
counter++;
}
}
}
}
}
}
return res;
}
}
动态规划
学习官方的题解。方法非常巧妙。
利用了哈希表和动态规划,不过这次不是用dp直接求解,而是间接通过更新dp数组(哈希表数组)求解。
哈希表种存储的是差值。每个等差数列的d是唯一的。
class Solution {
public int numberOfArithmeticSlices(int[] nums) {
int ans = 0;
int n = nums.length;
Map<Long, Integer>[] f = new Map[n];
for (int i = 0; i < n; ++i) {
f[i] = new HashMap<Long, Integer>();
}
for (int i = 0; i < n; ++i) {
for (int j = 0; j < i; ++j) {
long d = 1L * nums[i] - nums[j];
int cnt = f[j].getOrDefault(d, 0);
ans += cnt;
f[i].put(d, f[i].getOrDefault(d, 0) + cnt + 1);
}
}
return ans;
}
}