题目描述
给定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个整数,中间用空格分隔,每个数的绝对值不超过100。当K为0时,输入结束,该用例不被处理。
输出
对每个测试用例,在1行里输出最大和、最大连续子序列的第一个和最后一个元素,中间用空格分隔。如果最大连续子序列不唯一,则输出序号i和j最小的那个(如输入样例的第2、3组)。若所有K个元素都是负数,则定义其最大和为0,输出整个序列的首尾元素。
我们可以设置结构体数组dp,数组中存储dp值和此处序列首索引和尾索引,再根据状态转移方程来判断如何更新dp数组
代码如下:
#include <bits/stdc++.h>
using namespace std;
#define N 10005
int k;
struct node{ //利用s和e存储该序列首索引和尾索引
int v;
int s,e;
}dp[N];
int T[N];
int main(){
while(scanf("%d",&k) != EOF, k){
bool flag = false; //用flag标记输入数据是否都为负
for(int i=0;i<k;i++){
scanf("%d",&T[i]);
if(T[i] > 0) flag = true;
}
if(flag){
dp[0].v = T[0];
dp[0].s = dp[0].e = 0;
for(int i=1;i<k;i++){
if(dp[i-1].v + T[i] < T[i]){
dp[i].s = dp[i].e = i;
dp[i].v = T[i];
}else{
dp[i].s = dp[i-1].s;
dp[i].e = i;
dp[i].v = dp[i-1].v + T[i];
}
}
int MAX = dp[0].v, u = 0;
for(int i=1;i<k;i++){
if(dp[i].v > MAX){
MAX = dp[i].v;
u = i;
}
}
printf("%d %d %d\n",dp[u].v,T[dp[u].s],T[dp[u].e]);
}else{
printf("0 %d %d\n",T[0],T[k-1]);
}
}
return 0;
}