题意:给n个正整数,正整数s。求出1 -- > n之中最小的连续序列,使得他们的和大于s。
这一题厚白书上有详细的推导过程,我来捋一捋。若直接用二重循环,一个数一个数的判断,时间复杂度太高。可以使用前缀技巧,求出A[n]为前n个数之和。
这一下我们用二重循环就方便多啦,从1开始到i,找出最大的j 使得 a[i] - a[j] >= s。可是时间复杂度也挺高。
因为都是正整数,那么A数组肯定是递增的,那么我们就可以用二分查找了。
树上给的最优解是,用滑动窗口的办法。若A[i] - A[j] >= s那么j++,否则 i++。这样就形成了i-->j 的窗口,我们维护窗口就行了。为什么可以这样?因为A数列是递增的啊。
二分:
#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn = 100000 + 7;
int num[maxn];
long long A[maxn];
int main()
{
int n, s;
while(scanf("%d%d", &n, &s) == 2)
{
A[0] = 0;
for(int i = 1; i <= n; i++)
{
scanf("%d", &num[i]);
A[i] = A[i - 1] + num[i];
}
int ans = n + 1;
for(int i = 1; i <= n; i++)
{
int L = 0, R = i;
int mid = (L + R) / 2;
int t = A[i] - s;
while(L <= R)
{
if(A[mid] > t) R = mid - 1;
else L = mid + 1;
mid = (R + L) / 2;
}
//printf("%d\n", mid);
if(t >= 0) ans = min(ans, i - L + 1);
}
printf("%d\n", ans == n + 1 ? 0 : ans);
}
return 0;
}
滑动窗口:
#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn = 100000 + 7;
int num[maxn];
long long A[maxn];
int main()
{
int n, s;
while(scanf("%d%d", &n, &s) == 2)
{
A[0] = 0;
for(int i = 1; i <= n; i++)
{
scanf("%d", &num[i]);
A[i] = A[i - 1] + num[i];
}
int ans = n + 1, j = 1;
for(int i = 1; i <= n; i++)
{
if(A[i] - A[j] >= s)
{
while(A[i] - A[j] >= s && j < i) j++;
int t = i - j + 1;
ans = min(ans, t);
}
}
printf("%d\n", ans == n + 1 ? 0 : ans);
}
return 0;
}