题目链接
给定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( < 10000 ),第2行给出K个整数,中间用空格分隔。当K为0时,输入结束,该用例不被处理。
对每个测试用例,在1行里输出最大和、最大连续子序列的第一个和最后一个元 素,中间用空格分隔。如果最大连续子序列不唯一,则输出序号i和j最小的那个(如输入样例的第2、3组)。若所有K个元素都是负数,则定义其最大和为0,输出整个序列的首尾元素。
非常经典的 DP 题,找最大和其实很简单,坑点在于输出首尾元素,我们要注意从后往前推首元素时,当某个元素为 0 时,也算在内,举个例子:
5 0 9 -9 0 9应该输出:
9 0 9注意当最大和小于 0 时直接输出首位和末位元素即可,AC代码如下:
#include<bits/stdc++.h> using namespace std; typedef long long ll; int n,dp[10005],ans,a[10005],s,t,sum; int main(){ while(~scanf("%d",&n)&&n){ for(int i=0;i<n;i++) scanf("%d",&a[i]); memset(dp,0,sizeof(dp)); ans=-1e9; dp[0]=a[0]; for(int i=0;i<n;i++){ dp[i]=max(a[i],dp[i-1]+a[i]); if(dp[i]>ans) ans=dp[i],t=i; } s=t,sum=ans; while(s>=0){ if(sum-a[s]>=0) sum-=a[s],s--; if(sum==0&&a[s]) break; } if(ans>=0) printf("%d %d %d\n",ans,a[s+1],a[t]); else printf("0 %d %d\n",a[0],a[n-1]); } }