HDU 1231 最大连续子序列
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
思路
与luoguP1115 最大子段和相似,不同的是此题需要输出最大序列的首元素和尾元素。将前面各项累加,如果小于0,则置0,因为小于0,加上后面的数,会使后面的数变小,每次更新最大值。
我们还需要另外定义4个变量,begin、end、beginm、endm、分别用于储存循环过程中每次累加和的首尾元素下标和最大值的首尾元素下标。求和过程中每加一个数将end更新为所加数的下标,当重新计数时需要将begin和end都更新为开始元素。每次更新最大值是亦需更新下标。
代码
#include<iostream>
#include<cstdio>
using namespace std;
int w[10005];
int main()
{
int k;
while(1)
{
cin>>k;
if(k==0)
return 0; //k=0直接返回
for(int i=1;i<=k;i++)
scanf("%d",&w[i]);
int sum=w[1]; //用第一个数据赋初值
int max=sum;
int begin=1,end=1,beginm=1,endm=1; //begin和end都指向第一个数
for(int i=2;i<=k;i++)
{
if(sum>=0) //继续加 注意此处有等于,因为题中要求i和j最小
{
sum+=w[i];
end=i; //又加了一个数,更新end
}
else
{
sum=w[i]; //sum小于0,加了sum会变小
begin=i; //begin和end从计
end=i;
}
if(sum>max) //更新最大值
{
beginm=begin;
endm=end; //beginm和endm也需要更新
max=sum;
}
}
if(max<0) //题中要求 注意不能有等于
{
max=0;
beginm=1;
endm=k;
}
printf("%d %d %d\n",max,w[beginm],w[endm]);
}
return 0;
}