乍一看不是一道难题,但是写了我好几个小时;
首先第一想法用深搜,在VS上调试的时候,发现深搜好像不容易解决这种连续数组问题;只能得出所有情况,包括不连续的数组元素组合;
后来想到用双指针,但是超时了;
再后来,答案用的滑动窗口,可是个人感觉和双指针区别好像不是很大,不太懂为什么;
先贴一个深搜代码吧,对不要求连续元素的时候还是有用的,并且自己还输出了所有可能的情况:
#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
int num = 0;
vector<vector<int>>res_path;
vector<int>path;
void fun(vector<int>& nums, int k, int result, int index)
{
if (index < nums.size())
{
if (result < k)
{
num++;
res_path.push_back(path);
}
else
return;
}
index += 1;
for (int i = index; i < nums.size(); i++)
{
result = result * nums[i];
path.push_back(nums[i]);
fun(nums, k, result, i);
path.pop_back();
result = result / nums[i];
}
return;
}
int numSubarrayProductLessThanK(vector<int>& nums, int k) {
if (k == 0)return 0;
vector<int>nums1(nums);
sort(nums1.begin(), nums1.end());
if (k < nums1[0])return 0;
for (int i = 0; i < nums.size(); i++)
{
path.push_back(nums[i]);
fun(nums, k, nums[i], i);
path.pop_back();
}
return num;
}
int main()
{
int n;
cin >> n;
vector<int>nums(n);
for (int i = 0; i < n; i++)
{
cin >> nums[i];
}
int k;
cin >> k;
numSubarrayProductLessThanK(nums, k);
cout << num;
for (int i = 0; i < res_path.size(); i++)
{
cout << endl;
for (int j = 0; j < res_path[i].size(); j++)
{
cout << res_path[i][j] << " ";
}
}
return 0;
}
再贴一下双指针和滑动窗口,做对比:
双指针:
滑动窗口:
不太理解,两种方法都是 两个循环嵌套,为什么双指针不对呢?难道出错了吗,可是输了几个例子是对的。。。