给定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 Huge input, scanf is recommended.
代码如下
#include <iostream>
#include <algorithm>
using namespace std;
int main(){
int tt;
int a[10005];
int start,end;
while(cin >> tt){
if(tt == 0) break;
for(int i=1;i<= tt;i++)
cin >> a[i];
int maxhere=a[1],ans=a[1];
start = a[1];
int k=1;
for(int i=2;i <= tt;i++){
if(maxhere <= 0)
maxhere = a[i];
else
maxhere += a[i];
if(maxhere > ans){
ans = maxhere;
k = i;
}
}
int cal=0;
for(int i=k;i >= 1;i--){
cal += a[i];
if(cal == ans){
start = a[i];
break;
}
}
if(ans < 0) cout << "0 " << a[1] << " " << a[tt] << endl;
else
cout << ans << " " << start << " " << a[k] << endl;
}
return 0;
}
也可以使用动态规划。