相对于前面的43道题,这道题的难度可以排在前8名了。本来在数组中找最大子数组就比较难以解决了,这题需要找到和为特定值的子数组(或者最接近特定值)。
为了贪图简单,我第一种想到的方法是求出任意两个点之间的和,然后找最接近目标的。为了避免重复计算,填写一张n*n的二维数组,代表由i到j的子数组的和,以空间换时间。结果空间超限、时间超限。
第二种方法和第一种方法类似,二重循环遍历,暴力求解,但不再用表格存放中间结果,因此有很多重复计算。做了一些剪枝,比如从i开始的子数组,当末尾遍历到j时,sum大于目前的最接近目标值,那么跳出当前的第二重循环,因为再加下去只会更大。当然了,还是会超时的。
第三种方法有些奇怪,说不上有什么规矩或道理,但是通过了。head和tail都从0开始,代表子数组下标。head不动,由0到n遍历tail,遍历过程中更新sum,当sum>=M时,进入判断,判断sum与当前已有的最接近目标值的大小,然后做出相应的更新。此时,head就需要变化了,不然tail继续向后只会越加越大。此时进入循环,当前和sum减去head所在位置的值,然后head++,直到当前和小于M,在head向前移动的过程中,也需要判断sum与当前已有的最接近目标值的大小,做相应更新。和第二种方法优化的地方可能在于,它省略了很多重复求和的过程,因此通过了。
上述主要循环的代码如下:
int min = 100000000;
int head = 0, tail = 0, sum = 0;
for (int tail = 0; tail < N; tail++)
{
sum += value[tail]; //当sum小于当前最接近目标值的值min时,tail一直后移
if (sum >= M) //一旦sum大于M,就要进行判断了
{
if (sum < min)
{
min = sum;
solutions.clear(); //solutions是存放输出结果的vector<int>,当有更接近目标值的值出现时,要清空它
solutions.push_back(head);
solutions.push_back(tail);
}
else if (sum == min)
{
solutions.push_back(head);
solutions.push_back(tail);
}
while (sum >= min) //一旦sum大于M,就需要移动head,直到子数组的sum小于M,这里虽然条件是sum<min时跳出,但实际上是sum<M,因为当sum<min时,在循环内部min也会做出更新
{
sum -= value[head];
head++;
if (sum == min)
{
solutions.push_back(head);
solutions.push_back(tail);
}
if (sum < min && sum >= M)
{
min = sum;
solutions.clear();
solutions.push_back(head);
solutions.push_back(tail);
}
}
}
}
第四种方法是我参考其他人的代码才知道的方法,也是我目前觉得最有道理的一种方法,二分法。先求出任意点到零点的连续和,求和后数组呈递增状态,因此可以使用二分法找最接近目标的值。
二分法的详细代码推荐这位的博客:http://www.voidcn.com/article/p-ycvrjtcn-gu.html
最后,我使用的是第三种方法AC的,贴上第三种方法的代码:
#include<stdio.h>
#include<vector>
using namespace std;
int value[100000];
vector<int> solutions;
int main()
{
int N, M;
scanf("%d%d", &N, &M);
for (int i = 0; i < N; i++)
{
scanf("%d", &value[i]);
}
int min = 100000000;
int head = 0, tail = 0, sum = 0;
for (int tail = 0; tail < N; tail++)
{
sum += value[tail];
if (sum >= M)
{
if (sum < min)
{
min = sum;
solutions.clear();
solutions.push_back(head);
solutions.push_back(tail);
}
else if (sum == min)
{
solutions.push_back(head);
solutions.push_back(tail);
}
while (sum >= min)
{
sum -= value[head];
head++;
if (sum == min)
{
solutions.push_back(head);
solutions.push_back(tail);
}
if (sum < min && sum >= M)
{
min = sum;
solutions.clear();
solutions.push_back(head);
solutions.push_back(tail);
}
}
}
}
for (int i = 0; i < solutions.size() / 2; i++)
{
printf("%d-%d\n", solutions[2 * i] + 1, solutions[2 * i + 1] + 1);
}
return 0;
}