POJ 3061 Subsequence（尺取法）

题目链接: 传送门

Subsequence
Time Limit: 1000MS     Memory Limit: 65536K

题目描述

给定长度为n的数列整数以及整数S。求出总和不小于S的连续子序列的长度的最小值。如果解不存在，则输出0。

思路

O(nlogn)算法

#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;

int main()
{
	int T;
	scanf("%d",&T);
	while (T--)
	{
		int N,S,i;
		int ans[100005] = {0},sum[100005] = {0};
		scanf("%d%d",&N,&S);
		int res = N;
		for (i = 0;i < N;i++)
		{
			scanf("%d",&ans[i]);
			sum[i + 1] = sum[i] + ans[i];
		}
		if (sum[i] < S)
		{
			res = 0;
		}
		else
		{
			for (i = 0;sum[i] + S < sum[N];i++)
			{
				int tmp = lower_bound(sum + i,sum + N,sum[i] + S) - sum;
				res = min(res,tmp - i);
			}
		}
		printf("%d\n",res);
	}
	return 0;
} 

O(n)

#include<bits/stdc++.h>
using namespace std;

int main()
{
	int T;
	scanf("%d",&T);
	while (T--)
	{
		int N,S;
		int ans[100005] = {0};
		
		scanf("%d%d",&N,&S);
		for (int i = 0;i < N;i++)
		{
			scanf("%d",&ans[i]);
		}
	
		int res = N + 1;
		int sum = 0,t = 0,s = 0;
		for (;;)
		{
			while (t < N && sum < S)
			{
				sum += ans[t++];
			}
			if (sum < S)
			{
				break;
			}
			res = min(res,t - s);
			sum -= ans[s++];
		}
		
		if (res > N)
		{
			res = 0;
		}
		printf("%d\n",res);
	}
	return 0;
}