Time Limit: 1000MS | Memory Limit: 65536KB | 64bit IO Format: %I64d & %I64u |
Description
A sequence of N positive integers (10 < N < 100 000), each of them less than or equal 10000, and a positive integer S (S < 100 000 000) are given. Write a program to find the minimal length of the subsequence of consecutive elements of the sequence, the sum of which is greater than or equal to S.
Input
The first line is the number of test cases. For each test case the program has to read the numbers N and S, separated by an interval, from the first line. The numbers of the sequence are given in the second line of the test case, separated by intervals. The input will finish with the end of file.
Output
For each the case the program has to print the result on separate line of the output file.if no answer, print 0.
Sample Input
2 10 15 5 1 3 5 10 7 4 9 2 8 5 11 1 2 3 4 5
Sample Output
2 3
题意:给定长度为n的整数数列a0,a1,…,an-1以及整数S。求出总和不小于S的连续子序列的长度的最小值。如果解不存在,输出0。
思路:
1. 可以用O(n)的时间算好前缀和,之后就能在O(1)的时间计算区间上的总和。在确定起点s后,二分快速找出使得序列和不小于S的最近的终点就好了。算法复杂度是O(nlogn).
2. 第二种方法是利用尺取法求解:
(1).以s=t=sum=0初始化。
(2).只要依然有sum<S,就将sum增加ai,并将t增加1。
(3).如果(2)中无法满足sum>=S则终止。否则的话,更新res=min(res,t-s).
(4).将sum减去as,s增加1后回到(2).
算法的复杂度是O(n)。
前缀和+二分搜索
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int MAXN = 100000 + 1000;
int num[MAXN];
int sum[MAXN];
int n, s;
int main()
{
int casen;
scanf("%d", &casen);
while (casen--)
{
scanf("%d%d", &n, &s);
for (int i = 0; i < n; i++)
scanf("%d", &num[i]);
sum[0] = num[0];
for (int i = 1; i < n; i++)
sum[i] = sum[i - 1] + num[i];
if (sum[n - 1] < s)
{
printf("0\n");
continue;
}
int ret = n;
int b, e;
for (b = 0; sum[b] + s <= sum[n-1]; b++)
{
e = lower_bound(sum + b, sum + n, sum[b] + s) - sum;
ret = min(ret, e - b);
}
printf("%d\n", ret);
}
}
尺取法
#include<cstdio>
#include<cstring>
using namespace std;
#define min(a,b) a<b?a:b
const int MAXN = 100000 + 1000;
int num[MAXN];
int n, S;
int main()
{
int casen;
scanf("%d", &casen);
while (casen--)
{
scanf("%d%d", &n, &S);
for (int i = 0; i < n; i++)
scanf("%d", &num[i]);
int ret = n + 1;
int s = 0, t = 0, sum = 0;
for (;;)
{
while (t < n && sum < S)
sum += num[t++];
if (sum < S) break;
ret = min(ret, t - s);
sum -= num[s++];
}
if (ret>n)
ret = 0;
printf("%d\n", ret);
}
}