这道题一共产生了三种解决方法,最终实现全部测试点通过的是最后一种,现在每一种都说说吧。
- 暴力一点的方法
这种暴力的方法是最容易想到的,也就是遍历每一个元素,然后每一个元素向后累加,直到加出一个大于等于目标值为止,再进行下一步是否是可用结果的判断。
很明显,这种方法很容易超时,事实也是,有两个还是三个测试点超时了,因此我再考虑了第二种方法。 - 动态规划的方法
第一种方法超时的原因,一部分是来自于重复计算,比如{1,2,3,4,5,6}这个序列中,当分别计算1-4、2-4和3-4时,每次都会重复计算3+4,这样的重复计算会带来额外的开销,动态规划就是用来解决这一问题的,本质上就是用空间换时间,方法就是设计一个大的矩阵,用来保存每个子问题的结果,比如先计算出相差为1的两个量的和,基于这些和,再计算相差为2的和,如此类推。
理论上来说这样是没问题的,可是这需要巨大的空间来实现,题目中钻石数量最高有100000颗,C语言的数组无法满足这一要求,因此作罢。 - 二分查找的方法
该方法其实是对第一种和第二种的改进,主要解决两个问题:查找时的复杂度和子问题减少。
查找复杂度通过二分查找降低,子问题减少即钻石不用单个价值来保存,而是用其前面所有钻石价值和来保存,比如上例中,{1,2,3,4,5,6}不存储为这样,而是存储为{1,3,6,10,15,21},当需要某一段(如i到j)的价值时,只需要用j的价值减去第i-1个价值即可。
实现代码如下:
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
vector<int> sum, res;
int maxN = 999999999;
int N, K;
int search(int p)
{
int start = p, end = N;
while(start < end)
{
int mid = (start+end)/2;
if(sum[mid] - sum[p-1] >= K)
end = mid;
else
start = mid+1;
}
int temp = sum[start] - sum[p-1];
if(temp>=K && temp <= maxN)
{
if(temp<maxN)
{
maxN = temp;
res.clear();
}
res.push_back(p);
res.push_back(start);
}
return 0;
}
int main()
{
cin >> N >> K;
sum.resize(N+1);
sum[0] = 0;
for(int i=1;i<=N;i++)
{
cin >> sum[i];
sum[i] += sum[i-1];
}
for(int i=1;i<=N;i++)
search(i);
for(int i=0;i<res.size();i+=2)
cout << res[i] << "-" << res[i+1] << endl;
return 0;
}
有趣的是,之前我将maxN设的不够大,测试点2总是会报错,让我排查了好久这个问题,也就是说这个测试点会产生极大的和,因此这个值必须足够大。当然,还有更聪明的方法,那就是直接将这个值设为所有数的和,即sum[N]。
学无止境啊~