题目描述
给定 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( <100000) ,第 2 行给出 K 个整数,每个整数的范围-10000至10000 ,中间用空格分隔。
输出
对每个测试用例, 在 1 行里输出最大和、 最大连续子序列的第一个和最后一个元素的下标,中间用空格分隔。 如果最大连续子序列不唯一, 则输出序号 i 和 j 最小的那个(如输入样例的第 2、3组)。若所有 K 个元素都是负数,则定义其最大和为0,输出"0 0 0"。
样例输入
8
6 -2 11 -4 13 -5 -2 10
20
-10 1 2 3 4 -5 -23 3 7 -21 6 5 -8 3 2 5 0 1 10 3
8
-1 -5 -2 3 -1 0 -2 0
4
-1 -2 -4 -3
样例输出
27 0 7
27 10 19
3 3 3
0 0 0
思路
采用动态规划的方法,推出一般表达式为:
m
a
x
s
u
m
[
i
]
=
m
a
x
{
m
a
x
s
u
m
[
i
−
1
]
+
n
u
m
[
i
]
,
n
u
m
[
i
]
}
maxsum[i]=max\{maxsum[i-1]+num[i], num[i]\}
maxsum[i]=max{maxsum[i−1]+num[i],num[i]}
表示当最大序列包括第i个元素时,最大值maxsum[i]取值要么是第i-1个元素的最大值maxsum[i-1]加i元素本身,要么就是i元素,在这之中取一个最大值。
代码
#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
const int N=100010;
int dp[N];
int main(){
int k;
int num[100010];
while(scanf("%d", &k)!=EOF){
int sum=0, a=0, b=0; //ab纪录和的最大值的左右下标
for(int i=0; i<k; i++){
scanf("%d", &num[i]);
}
int maxn=0, p, q; //pq为当前和的左右下标,maxn纪录一个最大的和
for(int i=0; i<k; i++){
sum=max(sum+num[i], num[i]);
if(sum==num[i]){
p=q=i;
}else{
q=i;
}
if(sum>maxn){
maxn=sum;
a=p;
b=q;
}
}
printf("%d %d %d\n", maxn, a, b);
}
return 0;
}
此题目还有其他方法,详情可参照 最大连续子序列和:动态规划经典题目.