POJ 3061 Subsequence 尺取法

一、题目大意

给我们一个长度为N的数组和一个数字S,求数组中最少连续多少位求和能大于S,如果数组中连续元素求和不可能大于S,输出0。

二、解题思路

用尺取法,记录两个指针left和right,用来表示数组 [left,right) 区域内(左闭右开)内求和作为答案。

一开始left和right都取1,记录[left,right)的和sum,当left>right时跳出循环,在每一次循环中执行如下操作。

1、如果sum=S或者right>N
2、如果第1部循环结束,sum=S,更新res=min(res,right-left)
4、放大left回到1

这样不断执行,最终可以在O(N)时间内计算出答案

三、代码

#include 
using namespace std;
int N, S, arr[100007], res = 0;
void input()
{
    scanf("%d%d", &N, &S);
    res = N + 1;
    for (int i = 1; i <= N; i++)
    {
        scanf("%d", &arr[i]);
    }
}
void solve()
{
    // 我们的有效区间在[left,right)
    // 1 2 3 4 5,一开始没有有效区间
    int left = 1, right = 1, sum = 0;
    while (left <= right)
    {
        while (sum < S && right <= N)
        {
            sum += arr[right++];
        }
        if (sum < S)
        {
            return;
        }
        res = min(right - left, res);
        sum -= arr[left++];
    }
}
int main()
{
    int t = 0;
    scanf("%d", &t);
    while (t--)
    {
        input();
        solve();
        printf("%d\n", res == N + 1 ? 0 : res);
    }
    return 0;
}

你可能感兴趣的:(算法)