题目描述
给定 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
思路
由于数据量比较大,因为使用动态规划
dp[i]:包括下标i(以a[i]为结尾)的最大连续子序列和为dp[i]。
递推公式
dp[i]只有两个方向可以推出来:
- dp[i - 1] +a[i],即:a[i]加入当前连续子序列和
- a[i],即:从头开始计算当前连续子序列和
一定是取最大的,所以dp[i] = max(dp[i - 1] + a[i], a[i]);
dp数组的初始化
从递推公式可以看出来dp[i]是依赖于dp[i - 1]的状态,dp[0]就是递推公式的基础。
dp[0].val=a[0]
遍历顺序
递推公式中dp[i]依赖于dp[i - 1]的状态,需要从前向后遍历。
记录坐标
由于本题要求输出坐标位置,因此将dp定义为结构体用于记录坐标值
#include<bits/stdc++.h>
using namespace std;
//采用dp
const int N=100005;
struct {
int l;
int r;
int val;
}dp[N];
int a[N];
int main(void){
int k;
while(scanf("%d",&k)!=EOF){
for(int i=0;i<k;i++){
scanf("%d",&a[i]);
}
int res=a[0],l=0,r=0;
dp[0].val=a[0];
dp[0].l=0;
dp[0].r=0;
for(int i=1;i<k;i++){
if(dp[i-1].val+a[i]>=a[i]){
dp[i].val=dp[i-1].val+a[i];
dp[i].l=dp[i-1].l;
dp[i].r=i;
}else{
dp[i].val=a[i];
dp[i].l=i;
dp[i].r=i;
}
if(res<dp[i].val){
res=dp[i].val;
l=dp[i].l;
r=dp[i].r;
}
}
if(res<0){
printf("0 0 0\n");
}else{
printf("%d %d %d\n",res,l,r);
}
}
return 0;
}
本次思路来自于代码随想录——最大子序和