[HDU 1231 最大连续子序列]

题目连接  http://acm.hdu.edu.cn/showproblem.php?pid=1231

最大连续子序列

Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 65536/32768 K (Java/Others)
Total Submission(s): 23403    Accepted Submission(s): 10479

Problem Description

给定K个整数的序列{ N1, N2, ..., NK }，其任意连续子序列可表示为{ Ni, Ni+1, ...,
Nj }，其中 1 <= i <= j <= K。最大连续子序列是所有连续子序列中元素和最大的一个，
例如给定序列{ -2, 11, -4, 13, -5, -2 }，其最大连续子序列为{ 11, -4, 13 }，最大和
为20。
在今年的数据结构考卷中，要求编写程序得到最大和，现在增加一个要求，即还需要输出该
子序列的第一个和最后一个元素。

 

Input

测试输入包含若干测试用例，每个测试用例占2行，第1行给出正整数K( < 10000 )，第2行给出K个整数，中间用空格分隔。当K为0时，输入结束，该用例不被处理。

 

Output

对每个测试用例，在1行里输出最大和、最大连续子序列的第一个和最后一个元
素，中间用空格分隔。如果最大连续子序列不唯一，则输出序号i和j最小的那个（如输入样例的第2、3组）。若所有K个元素都是负数，则定义其最大和为0，输出整个序列的首尾元素。

 

Sample Input

  
  

   
   6
-2 11 -4 13 -5 -2
10
-10 1 2 3 4 -5 -23 3 7 -21
6
5 -8 3 2 5 0
1
10
3
-1 -5 -2
3
-1 0 -2
0
  
  

 

Sample Output

  
  

   
   20 11 13
10 1 4
10 3 5
10 10 10
0 -1 -2
0 0 0


   
   

    
    

     
     Hint
    
    
Hint
   
   
   
    
Huge input, scanf is recommended.
  
  

 

Source

浙大计算机研究生复试上机考试-2005年

 

简单的题意，开始没多想，用sum[i]求出从1到i的和，然后用2个for遍历。

当时犹豫了一会，题目有个Hint： Huge input, scanf is recommended.

好像有很多测试数据，抱着侥幸的心理提交果然TLE了。。

下面是TLE代码。

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
int n;
int s[10010];
int sum[10010];
int main()
{
//	freopen("input.txt", "r", stdin);
	while (scanf("%d", &n) != EOF)
	{
		if (!n)
			break;
		int r = 1, l = n, ans = -1;
		s[0] = 0;
		memset(sum, 0, sizeof(sum));
		for (int i = 1; i <= n; i++)
		{
			scanf("%d", &s[i]);
			sum[i] += sum[i - 1] + s[i];
		}
		for (int i = 0; i < n; i++)
		{
			for (int j = i + 1; j <= n; j++)
			{
				if (sum[j] - sum[i] > ans)
				{
					ans = sum[j] - sum[i];
					r = i + 1;
					l = j;
				}
			}
		}
		if (ans == -1)
			ans = 0;
		printf("%d %d %d\n", ans, s[r], s[l]);
	}
	return 0;
}

这个算法为O(n^2),因为测试样例多，超时了(以后做题不仅要考虑单个样例的耗时，还要考虑样例组数)

后来仔细想了想。能否在预处理出数组sum的同时，直接处理出答案？

如样例 1 2 3 4 -5        当sum加到-5时，发现sum减少了，

如样例 1 2 3 4 -5 6     虽然sum减少了，但是继续往后加，sum还是在增加

如样例 1 2 3 4 -11 6    这个时候加到-11为止，sum是负数，明显这个-11不能加，越过这个-11，就要从零开始，这个时候比较一下1+2+3+4和6的大小就能得出答案，可以用变量max记录，

这样就是O(n)级了，题目要求输出的起始位置和终点位置只要在比较max的时候记录一下就可以了

187ms过的

代码。。

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
#define INF 0x3f3f3f3f
int n;
int s[10010];
int main()
{
//	freopen("input.txt", "r", stdin);
	while (scanf("%d", &n) != EOF)
	{
		if (!n)
			break;
		int r = 1, l = n;
		int r1 = 1, l1 = n;
		int max, sum = 0;
		bool flag = false;
		for (int i = 1; i <= n; i++)
			scanf("%d", &s[i]);
		max = -1;
		for (int i = 1; i <= n; i++)
		{
			sum += s[i];
			l1 = i;
			flag = false;
			if (sum < 0)
			{
				sum = 0;
				l1--;
				r1 = i + 1;
				flag = true;
			}
			if (sum > max && !flag)
			{
				max = sum;
				r = r1;
				l = l1;
			}
		}
		printf("%d %d %d\n", (max == -1 ? 0 : max), s[r], s[l]);
	}
	return 0;
}