UVA之1121 - Subsequence

【题目】

A sequence of N positive integers (10 < N < 100 000), each of them less than or equal 10000, and a positive integer S (S <
100 000 000) are given. Write a program to find the minimal length of the subsequence of consecutive elements of the sequence, the sum of which is greater than or equal to S.

Input

Many test cases will be given. For each test case the program has to read the numbers N and S, separated by an interval, from the first line.
The numbers of the sequence are given in the second line of the test case, separated by intervals. The input will finish with the end of file.

Output

For each the case the program has to print the result on separate line of the output file. If there isn't such a subsequence, print 0 on a line by itself.

Sample Input

10 15

5 1 3 5 10 7 4 9 2 8

5 11

1 2 3 4 5

Sample Output

2

3

【分析】

本题最直接的思路是二重循环，枚举子序列的起点和终点。

代码例如以下（输入数据已存入数组A[1]~A[n]）。

int ans = n+1;

for(int i = 1; i <= n; i++)

  for(int j = i; j <= n; j++) {

    int sum = 0;

    for(int k = i; k <= j; k++) sum += A[k];

    if(sum >= S) ans = min(ans, j-i+1);

  }

printf("%d\n", ans == n+1 ? 0 : ans);

非常可惜，上述程序的时间复杂度是O(n3)的，因此，当n达到100 000的规模后，程序将无能为力。

有一个方法能够减少时间复杂度。即常见的前缀和技巧。

令Bi=A1+A2+…+Ai，规定B0=0，则能够在O(1)时间内求出子序列的值：Ai+Ai+1+…+Aj=Bj-Bi-1。

这样。时间复杂度降为O(n2)，代码例如以下。

B[0] = 0;

for(int i = 1; i <= n; i++) B[i] = B[i-1] + A[i];

int ans = n+1;

  for(int i = 1; i <= n; i++)

    for(int j = i; j <= n; j++)

      if(B[j] - B[i-1] >= S) ans = min(ans, j-i+1);

printf("%d\n", ans == n+1 ? 0 : ans);

遗憾的是，本题的数据规模太大。O(n2)时间复杂度的算法也太慢。

不难发现，仅仅要同一时候枚举起点和终点，时间复杂度不可能比O(n2)更低，所以必须另谋他路。比方，能否够不枚举终点。仅仅枚举起点。或者不枚举起点，仅仅枚举终点呢？

我们首先试试仅仅枚举终点。对于终点j，我们的目标是要找到一个让Bj-Bi-1≥S。且i尽量大（i越大。序列长度j-i+1就越小）的i值。也就是找一个让Bi-1≤Bj-S最大的i。考虑图1-29所看到的的序列。

当j=5时。B5=12，因此目标是找一个Bi-1≤12-7=5的最大i。注意到B是递增的（别忘了，本题中全部Ai均为整数），所以能够用二分查找。

【代码】

/*********************************

*   日期：2014-5-14

*   作者：SJF0115

*   题号: 1121 - Subsequence

*   地址：http://uva.onlinejudge.org/index.php?option=com_onlinejudge&Itemid=8&category=246&page=show_problem&problem=3562

*   来源：UVA

*   结果：Accepted

**********************************/

#include <iostream>

#include <stdio.h>

#include <algorithm>

#include <string.h>

using namespace std;

#define N 100001

int A[N];

int B[N];

//二分查找最接近target但不大于target

int BinarySerach(int target,int R){

    int L = 0;

    int mid = 0;

    while(L < R){

        mid = L + (R - L) / 2;

        if(B[mid] > target){

            R = mid;

        }

        else{

            L = mid + 1;

        }

    }

    return L;

}

int main(){

    int n,s,i,j;

    //freopen("C:\\Users\\wt\\Desktop\\acm.txt","r",stdin);

    while(scanf("%d %d",&n,&s) != EOF){

        int minLen = n+1;

        B[0] = 0;

        for(i = 1;i <= n;i++){

            scanf("%d",&A[i]);

            //序列前缀和

            B[i] = B[i-1] + A[i];

        }

        for(j = 1;j <= n;j++){

            int target = B[j] - s;

            //二分查找

            int index = BinarySerach(target,j-1);

            if(index > 0){

                minLen = min(minLen,j-index+1);

            }

        }

        //没有满足条件的序列

        if(minLen == n+1){

            minLen = 0;

        }

        cout<<minLen<<endl;

    }//while

    return 0;

}

【类似题目】

编程之美之2.14 求数组的子数组之和的最大值