序列 {a} ,求最短子序列,且S>=K (S为序列元素的和)a[i]>=0.
暴力如下O(n^2)
if(a[i]-a[j]>=K) ans=min(ans,i-j+1) i<j
变形一下 a[j]<=a[j]-K , 我们现在要找最大的这样的 j
注意a[i]>=0 s[i] 前缀和数组是单调序列,可以二分查找 j
#include <iostream>
#include <cstring>
#include <algorithm>
#include <vector>
using namespace std;
const int N=1e5+4;
#define int long long
int ans,k,a[N],n,s[N];
main(){
int i,j;
while(cin>>n>>k){
s[0]=0;
for(i=1;i<=n;i++) cin>>a[i],s[i]=s[i-1]+a[i];
ans=1<<30;
for(j=1;j<=n;j++){
i=lower_bound(s,s+j,s[j]-k)-s;
if(i>0) ans=min(ans,j-i+1);
}
cout<<(ans==1<<30?0:ans)<<'\n';
}
}