一开始二分,一直WA,怀疑是因为lower_bound,找不到结果时返回n+1。所以把答案%(n+1),然后就过了。
看大白书上说找不到就输出0,可是看了好几遍原题,上面没有说啊喂,怎么可以这样QAQ。
大白书上有O(n)的解法,就是枚举+查找的思路,查找时优化了一下使得总时间为O(n)。
典型的思路,枚举一个量,对另一个量进行查找,可以是lower_bound,O(nlogn),也可以根据递增以及求最小值的条件,直接用while更新到新的最优值O(n)。
之所以不用两个while是因为编码麻烦,容易出错而且不直观。最好是枚举一个,while另一个。
二分
#include<bits/stdc++.h>
#define maxn 100010
using namespace std;
typedef long long ll;
ll N,S;
ll a[maxn];
ll sum[maxn];
bool ok(ll K)
{
for(ll i=K;i<=N;i++)
if(sum[i]-sum[i-K]>=S) return true;
return false;
}
int main()
{
while(~scanf("%lld %lld",&N,&S))
{
for(ll i=1;i<=N;i++)
{
scanf("%lld",&a[i]);
sum[i]=sum[i-1]+a[i];
}
ll L,R;
L=0,R=N+1;
while(L<R)
{
ll M=L+((R-L)>>1);
if(ok(M)) R=M;
else L=M+1;
}
printf("%lld\n",L%(N+1));
}
return 0;
}
枚举+查找
#include<bits/stdc++.h>
#define maxn 100010
using namespace std;
int a[maxn];
int b[maxn];
int n,s;
int main()
{
while(~scanf("%d %d",&n,&s))
{
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
b[i]=b[i-1]+a[i];
}
int i=0;
int MIN=n+1;
for(int j=1;j<=n;j++)
{
if(b[j]-b[i]<s) continue;
while(i<j&&b[j]-b[i]>=s) i++;
MIN=min(MIN,j-i+1);
}
printf("%d\n",MIN%(n+1));
}
return 0;
}