Subsequence
题目大意:
N个正整数的序列(10 < N < 100 000),每个都小于或等于10000,并且给出一个正整数S (S < 100 000 000)。编写一个程序,找出最小长度的序列中连续元素的子序列,其总和大于或等于S。
很容易就可以想到暴力的 O(n3) 算法,可能会超时
不够快
稍微想一下又可以优化得到 O(n2) 算法,应该能过。
还是不够快
用二分法加上队列优化甚至可以得到一种 O(n*log2n) 的算法,对这道题来说已经足够了
有没有可能再快一点?
这道题的巧妙之处在于,经过一系列的优化之后,你甚至可以得到一种近似于O(n) 的算法:
先将将整个数组读入时使用 前缀和 先记录下来(最好使用快读,如果这道题数据范围再大一些就会引起超时),然后用两个指针,一个队头一个队尾,由队尾开始一边扫描到n,如果子序列和大小大于S的话,就将队头指针后移,因为队头一直递增,所以扫描一遍一定不会大于n,整体时间复杂度也就降到O(N)了。
完整代码如下:
#include <iostream>
#include <cstdio>
#include <cstring>
#include <cctype>
#include <algorithm>
using namespace std;
int Read(){//快读模板
char c;
int ans = 0;
c = getchar();
while(!isdigit(c))
c=getchar();
while(isdigit(c)){
ans = ans*10+c-48;
c = getchar();
}
return 0;
}
int main(){
while(cin>>n)//多组数据
{
memset(sum,0,sizeof(sum));//初始化数组
ans=def;
S=Read();
for(int i=1;i<=n;i++){//读入以及前缀和处理
a[i]=Read();
sum[i]=sum[i-1]+a[i];
}
int head=0,tail=1;
for(tail=1;tail<=n;tail++)
{
while(sum[tail]-sum[head]>=S)
{
if(sum[tail]-sum[head]>=S)
ans=min(ans,tail-head);
head++;
}
}
printf("%d\n",(ans==def)?0:ans);
}
return 0;
}