好久没更博客了,今天重新整理一下之前学习的一些技巧——尺取法。
在《挑程》中对于尺取法的定义如下:通常是指对数组保存一对下标(起点、终点),然后根据实际情况交替推进两个端点直到得出答案的方法,这种操作很像是尺取虫爬行的方式故得名。
用一个题目(POJ-3061)来具体解释一下:
问题描述:
分析:
1.暴力:
这题当然可以用暴力的方法求,但是很明显这样时间复杂度会非常大,就和之前的连续子序列和最大,用暴力枚举起点和终点,这样是O(n^2)。
2.另外一种思路:
由于所有的元素都大于零,如果子序列[s,t)满足as+…+at-1>=S,那么对于任何的t < t’一定有as+…+at-1 >= s。
对于区间[s,t)上的总和来说如果令sum(i) = a0 + a1 + … +ai-1
那么, as+as+1+…+at-1 = sum(t) - sum(s)。
因此预先以O(n)的时间计算好sum的话,就可以以O(1)的时间计算区间上的总和。这样子序列的起点s一旦确定以后,便可以用二分搜索快速确定使序列和不小于S的结尾t的最小值。则时间复杂度为O(nlogn)。
3.最后还有一个算法可以达到O(n),它就是尺取法:
我们设以as开始总和最初大于S时的连续子序列为as+…+at-1,这时as+1 + … + at-2 < as + … + at-2 < S
这样我们设计如下算法:
- 以s=t=sum=0初始化
- 只要依然有sum < S,sum就一直加at,并将t增加1。
- 如果第二步中无法满足sum>=S则终止。否则更新res=min(res, t-s)
- 将sum减去as, s增加1回到第二步。
代码如下:
#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn = 100000+10;
int main()
{
int T;
scanf("%d\n",&T);
while(T--)
{
int a[maxn];
int n,s;
scanf("%d%d",&n,&s);
for(int i=0; i<n; i++)
{
scanf("%d",&a[i]);
}
int crt = 0, last = 0;
int res = n + 1;
int sum = 0;
for(;;)
{
while(last < n && sum < s)
{
sum += a[last++];
}
if(sum < s) break;
res = min(res, last - crt);
sum -= a[crt++];
}
if(res > n)
res = 0;
printf("%d\n",res);
}
return 0;
}
这样说尺取法的过程就是如下四步:
1.初始化左右端点
2.不断扩大右端点,直到满足条件
3.如果第二步中无法满足条件,则终止,否则更新结果
4.将左端点扩大1,然后回到第二步