给定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,输出整个序列的首尾元素。
分析:关键问题在于如何设计状态和状态转移方程。
若是枚举左右端点,复杂度为O(N²),就算记录前缀和,利用差值找最优解,总体复杂度也为O(N²)。
我们考虑用一个结构体vector来记录以第x个元素结尾的最大连续和序列的和值以及序列起始元素号码。(因为要输出初始元素,故采用结构体记录,相当于简单问题中的dp[n])
考虑如何进行状态转移。要求当前元素为结尾的最长连续序列和,只有两种情况:第一种为以前一个元素为结尾的最长连续序列和的值大于0,则以当前元素为结尾的最长连续序列和=以前一个元素结尾的最长连续序列和+当前元素。 第二种为以前一个元素为结尾的最长连续序列和的值小于等于0,则以当前元素为结尾的最长连续序列和=当前元素。
状态转移方程在思路上简化表示为dp[i]=max{A[i],dp[i-1]+A[i]}
根据题意,计算顺序为从前往后,记录一下所求最大连续子序列的起始位置,再按照要求输出即可
题中红色字体部分可以在学习动态规划问题时简化掉,只是针对数据处理记录上的问题。
我的参考题解:
#include<iostream>
#include<vector>
using namespace std;
struct node{ //用结构体记录连续子序列的总和以及初始
int data;
int start;
};
int main(){
int K,max=-1,tag=-1;
cin>>K;
vector<int> v(K);
vector<node> ans(K);
for(int i=0;i<K;i++){
cin>>v[i];
}
ans[0].data=v[0];//初始条件
ans[0].start=0;
for(int i=1;i<K;i++){
if(ans[i-1].data>0){ //若以前一个元素结尾的最大子序列和为负,则以当前元素结尾的最大和子序列为自己本身
ans[i].data=ans[i-1].data+v[i];
ans[i].start=ans[i-1].start;
}else{ //若以前一个元素结尾的最大子序列和为非负
ans[i].data=v[i];
ans[i].start=i;
}
}
for(int i=0;i<K;i++){
if(ans[i].data>max){
max=ans[i].data;
tag=i;
}
}
if(tag==-1){
cout<<"0 "<<v[0]<<" "<<v[K-1];
}else{
cout<<max<<" "<<v[ans[tag].start]<<" "<<v[tag];
}
}
题目链接:https://www.nowcoder.com/questionTerminal/afe7c043f0644f60af98a0fba61af8e7