给定 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。
第 2 行给出 K 个整数 N1,…,NK。
输出格式
每组数据输出一行结果,包含最大子序列的和以及子序列的第一个下标 i 和最后一个元素的下标 j。
所有元素下标为 0∼K−1。
如果最大子序列不唯一,则选择 i 最小的那个子序列,如果仍不唯一,则选择 i 最小的子序列中 j 最小的那个子序列。
若所有 K 个元素都是负数,则定义其最大和为 0,输出 0 0 0。
数据范围
1≤K≤105,
−10000≤Ni≤10000,
输入最多包含 10 组数据。
输入样例:
8
6 -2 11 -4 13 -5 -2 10
5
10 -10 10 -10 10
8
-1 -5 -2 3 -1 0 -2 0
4
-1 -2 -4 -3
输出样例:
27 0 7
10 0 0
3 3 3
0 0 0
如果前i-1个最大连续子序列和小于0,则第i个最大连续子序列为p[i],否则为dp[i-1]+p[i],L数组存当前最长连续子序列左端点,注意L[1]应为1
#include<cstdio>
#include<algorithm>
using namespace std;
typedef long long ll;
const int N = 1e5+10;
int p[N],dp[N],L[N],n,ans;
int main(){
while(~scanf("%d",&n)){
bool flag(0);
for(int i(1);i<=n;i++){
scanf("%d",&p[i]);
if(p[i]>=0)flag=1;
if(dp[i-1]>=0){dp[i]=dp[i-1]+p[i];if(i>>1)L[i]=L[i-1];else L[i]=1;}
else dp[i]=p[i],L[i]=i;
}
int ans(-1e9),l,r;
for(int i(1);i<=n;i++){
if(ans<dp[i])r=i,l=L[i],ans=dp[i];
}
if(!flag) printf("0 0 0\n");
else {
printf("%d %d %d\n",ans,l-1,r-1);
}
}
}