LA 2678 – Subsequence

看到限时3S，自己写了一个二重循环的，然后华丽的 TLE。。。T T

瞄了瞄书上，作者的思路果然是很好。膜拜中。

他只枚举了终点，然后用二分查找。

用到了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);
	}
}