作者的思路果然是很好。膜拜中。
他只枚举了终点,然后用二分查找。
用到了lower_bound函数,这个lower_bound(first,last,val)在first和last中的前闭后开区间进行二分查找,返回大于或等于val的第一个元素位置。如果所有元素都小于val,则返回last的位置。
简单的说,就是当val存在时,返回它出现的第一个位置。如果不存在,就返回一个这样的下标i,在此处插入val(原来的元素sum[i],sum[i+1]…全部往后移动一位)后序列仍然有序。
而upper_bound就是当val存在时,返回它出现的最后一个位置。如果不存在,就返回一个这样的下标i,在此处插入val(原来的元素sum[i],sum[i+1]…全部往后移动一位)后序列仍然有序。
- #include<cstdio>
- #include<algorithm>
- using namespace std;
- const int MAXN=100000+10;
- int sum[MAXN];
- int main()
- {
- int n,s,t;
- while(scanf("%d%d",&n,&s)!=EOF)
- {
- sum[0]=0;
- for(int i=1;i<=n;i++)
- {
- scanf("%d",&t);
- sum[i]=sum[i-1]+t;
- }
- int ans=n+1;
- for(int i=1;i<=n;i++)
- {
- if(sum[i]-s>0)
- {
- int j=lower_bound(sum,sum+i,sum[i]-s)-sum;
- ans=min(ans,i-j+1);
- }
- }
- printf("%d\n",ans==n+1 ? 0:ans);
- }
- }
作者进一步优化的代码就更好了,因为i是递增的,sum也是递增的,所以sum[j]<=sum[i]-s也是递增的,就是说满足条件j的位置也是递增的
- #include<cstdio>
- #include<algorithm>
- using namespace std;
- const int MAXN=100000+10;
- int sum[MAXN];
- int main()
- {
- int n,s,t;
- while(scanf("%d%d",&n,&s)!=EOF)
- {
- sum[0]=0;
- for(int i=1;i<=n;i++)
- {
- scanf("%d",&t);
- sum[i]=sum[i-1]+t;
- }
- int ans=n+1,j=0;
- for(int i=1;i<=n;i++)
- {
- if(sum[i]-s<0)continue;
- while(sum[i]-sum[j]>=s) j++;
- ans=min(ans,i-j+1);
- }
- printf("%d\n",ans==n+1 ? 0:ans);
- }
- }