对《挑战程序设计竞赛》的一个记录
第三章 出类拔萃——中级篇
3.2 常用技巧精选
(1)尺取法
poj 3061 Subsequence
给定长度为n的数列整数a0,a1,…an-1以及证书S。求出总和不小于S的连续子序列的长度的最小值。如果解不存在在,则输出0.
已知:
10 < <script type="math/tex" id="MathJax-Element-61"><</script>n < <script type="math/tex" id="MathJax-Element-62"><</script>10^5
0 < <script type="math/tex" id="MathJax-Element-63"><</script>ai ≤ 10^4
S < <script type="math/tex" id="MathJax-Element-65"><</script> 10^8
sample input
n = 10
S = 15
a = {5,1,3,5,10,7,4,9,2,8}
sample output
2 (5 + 10)
这题比较好想的一个思路就是先求出前n项的和,再在满足sum>=S的时候,val = sum - S,二分查找之前的项中满足≤val的最大项。
例如:
index: | 0 | 1 | 2 | 3 | 4 | 5 | 6 | 7 | 8 | 9 |
---|---|---|---|---|---|---|---|---|---|---|
a : | 5 | 1 | 3 | 5 | 10 | 7 | 4 | 9 | 2 | 8 |
sum: | 5 | 6 | 9 | 14 | 24 | 31 | 35 | 44 | 46 | 54 |
第一个满足sum >= 15的值是下标为4的值,以此为例,前5项的和为24,val = 24 - 15 = 9,我要找前5项中sum<=9的最大小标,可以找到是小标为2的值,因此可得到连续子序列最小的长度为4 - 2 = 2 即5 + 10
这个算法的时间复杂度为:O( nlogn )
代码如下:
#include <iostream>
#include <cstdio>
#include <cmath>
#include <algorithm>
#define sf scanf
#define pf printf
using namespace std;
const int Maxn = 100010;
int T,n,s;
int sum[Maxn];
int solved(int l,int r,int k)
{
while(l <= r)
{
int mid = (l + r) / 2;
if(sum[mid] <= k)
l = mid + 1;
else
r = mid - 1;
}
return r;
}
int main()
{
int a;
sf("%d",&T);
while(T--)
{
sf("%d%d",&n,&s);
for(int i = 0;i < n;i ++)
{
sf("%d",&a);
if(i == 0) sum[i] = a;
else sum[i] = sum[i - 1] + a;
}
int Min = n + 1;
for(int i = 0;i < n;i ++)
{
if(sum[i] >= s)
Min = min(Min,i - solved(0,i,sum[i] - s));
}
pf("%d\n",Min > n?0:Min);
}
return 0;
}
那什么是尺取法?尺取法能更高效地解决此类问题。
我们设以 as 开始总和最初大于S时的连续子序列 as+...+at−1 ,这时
as+1+...+at−2<as+...+at−2<S
所以从 as+1 开始总和最初超过S的连续子序列如果是 as+1+...+at′−1 的话,则必然有 t≤t′