题目描述
给定K个整数的序列{ N1, N2, …, NK },其任意连续子序列可表示为{ Ni, Ni+1, …, Nj },其中 1 <= i <= j <= K。最大连续子序列是所有连续子序列中元素和最大的一个,例如给定序列{ -2, 11, -4, 13, -5, -2 },其最大连续子序列为{ 11, -4, 13 },最大和为20。现在增加一个要求,即还需要输出该子序列的第一个和最后一个元素。
输入描述:
测试输入包含若干测试用例,每个测试用例占2行,第1行给出正整数K( K< 10000 ),第2行给出K个整数,中间用空格分隔。当K为0时,输入结束,该用例不被处理。
输出描述:
对每个测试用例,在1行里输出最大和、最大连续子序列的第一个和最后一个元素,中间用空格分隔。如果最大连续子序列不唯一,则输出序号i和j最小的那个(如输入样例的第2、3组)。若所有K个元素都是负数,则定义其最大和为0,输出整个序列的首尾元素。
输入
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
输出
20 11 13
10 1 4
10 3 5
10 10 10
0 -1 -2
0 0 0
#include<iostream>
#include<cstdio>
using namespace std;
const int MAXN=10000;
int dp[MAXN];//dp[i]表示以第i个元素为末尾的连续序列的最大和,只有两种情况,一个或多个
int a[MAXN];
void maxSub(int n,int& Max,int& first,int& last){
Max=0,first=0,last=n-1;
//第一次遍历找出最大子序列和
for(int i=0;i<n;i++){
if(i==0) dp[i]=a[i];//初始
else dp[i]=max(dp[i-1]+a[i],a[i]);//多个vs一个
if(dp[i]>Max) Max=dp[i];
}
//第二次遍历确定最大子序列的两个边界
for(int i=0;i<n;i++){
if(i==0){
dp[i]=a[i];
}
else if(dp[i-1]+a[i]>a[i]){
last=i;//当前子序列继续扩充元素
dp[i]=dp[i-1]+a[i];
}
else{
first=i;//从当前的元素重新往后找,赋值first
dp[i]=a[i];
}
if(dp[i]==Max){
last=i;//此时已经找到最大的子序列,给last赋最终值
return;
}
}
}
int main(){
int n;
while(cin>>n){
for(int i=0;i<n;i++){
cin>>a[i];
}
int Max,first,last;
maxSub(n,Max,first,last);
cout<<Max<<" "<<a[first]<<" "<<a[last]<<endl;
}
}