PAT.A1007 Maximum Subsequence Sum

返回目录

题意

给一个数字序列ai,a2…an,求i,j(1≤i≤j≤n),使得ai+…+aj最大，输出最大和以及ai,aj。
如果有多种方案使得和最大，那么输出其中i,j最小的一组。
如果所有数都小于0，那么认为最大和为0，并输出首尾元素。

样例（可复制）

10
-10 1 2 3 4 -5 -23 3 7 -21

//output
10 1 4

注意点

1. 本题使用动态规划，问题为最大连续序列和。
2. 如果4号测试点没过的话检查一下输入全负数的情况。
3. 注意样例中的1和4不是下标，而是起点和终点的值

#include<bits/stdc++.h>
using namespace std;

const int maxn=10010;
int n,a[maxn],dp[maxn],s[maxn]={0};//a[i]存放序列，dp[i]存放到i为止的最大和，s存放i为止的最大和的起始点
bool flag=false;
int main(){
	cin>>n;
	for(int i=0;i<n;i++){
		scanf("%d",&a[i]);
		if(a[i]>=0)flag=true;
	}
	if(!flag){
		cout<<"0 "<<a[0]<<" "<<a[n-1]<<endl;
        return 0;
	}
	dp[0]=a[0];
	for(int i=1;i<n;i++){
		if(dp[i-1]+a[i]>a[i]){
			dp[i]=dp[i-1]+a[i];
			s[i]=s[i-1];
		}else{
			s[i]=i;
			dp[i]=a[i];
		}
	}
	int maxx=0;
	for(int i=1;i<n;i++){
		if(dp[maxx]<dp[i])maxx=i;
	}
	cout<<dp[maxx]<<" "<<a[s[maxx]]<<" "<<a[maxx]<<endl;
    return 0;
}