易保研机试训练营-基础营|动态规划|C -最大连续子序列

题意

给定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

Huge input, scanf is recommended.

超时代码

#include 
#include 
using namespace std;
struct info{
	int begin;
	//自己是终点
	int sum; 
};
int main() {
	int g[10000];
	info h[10000];
	int k;
	bool flag = false;
	while(cin >> k) {
		flag = false;
		if(k == 0)
			break;
		for(int i = 1; i <= k; ++i) {
			cin >> g[i];
			if(g[i] >= 0) flag = true;
			h[i].begin = i;
			h[i].sum = g[i];
			int s = h[i].sum;
			for(int j = i - 1; j >= 1; --j) {
				s += g[j];
				if(s > h[i].sum) {
					h[i].sum = s;
					h[i].begin = j;
				}
			}
		}
		info M;
		M.sum = h[1].sum;
		M.begin = h[1].begin;
		int end = 1;
		for(int i = 2; i <= k; ++i) {
			if(h[i].sum > M.sum) {
				M.sum = h[i].sum;
				M.begin = h[i].begin;
				end = i;
			}
		}
		if(flag == false)
			cout << 0 << ' ' << g[1] << ' ' << g[k] << endl;
		else {
			cout << M.sum << ' ' << g[M.begin] << ' ' << g[end] << endl;
		}
	}
	return 0;
} 

正确代码

#include 
using namespace std;
int main() {
	int g[10000];
	int k;
	bool flag = false;
	while(cin >> k) {
		flag = false;
		if(k == 0)
			break;
		for(int i = 1; i <= k; ++i) {
			cin >> g[i];
			if(g[i] >= 0) flag = true;
		}
		if(flag == false) {
			cout << 0 << ' ' << g[1] << ' ' << g[k] << endl;
			continue;
		}
		int MAX = g[1];
		int sum = g[1];
		int begin = 1;
		int end = 1;
		int temp = 1;
		for(int i = 2; i <= k; ++i) {
			if(sum < 0) {
				sum = 0;
				temp = i;
			}
			sum += g[i];
			if(sum > MAX) {
				MAX = sum;
				end = i;
				begin = temp;
			}
		}
		cout << MAX << ' ' << g[begin] << ' ' << g[end] << endl;
	}
	return 0;
}