Leetcode上有一个Max Sum问题,题目的描述很简单,即对于一个输入的数组,求其连续子数组的最大和。比如说,A={ 6,-1,5,4,-7},那么其连续子数组的最大和为6 + (-1) +5 + 4 = 14.
分析:
这是很简单的动态规划题,动态规划做多了,第一个想法就是用一个一维数组记录以每个元素为结尾的子序列的最大和,然后再扫描一遍这个数组,以获取最大值。但是仔细分析一下,其实不需要用一个一维数组记录每个位置的最大和,因为没有要求返回具有最大和的子序列。所以使用一个变量记住当前连续子序列的最大和就行了,再用另一个变量记住最大和max,当curMax更新时,max也随之更新就可以了。
int maxSum(int a[],int n)
{
if(n==0)
return 0;
int curMax=a[0],maxsum=a[0];
for(int i=1;i<n;i++)
{
curMax=max(a[i],curMax+a[i]);
maxsum=max(maxsum,curMax);
}
return maxsum;
}
在hdoj上有一个该问题的变体。
http://acm.hdu.edu.cn/showproblem.php?pid=1003
该题需要返回连续子序列的起始位置,因此需要用一维数组来记录以每个元素为结尾的子序列的最大和。假设dp[i]表示a[0]到a[i]的maxsum,则dp[i]=max(dp[i],dp[i-1]+a[i])。题目需要返回连续子序列的起始位置,所以在求maxsum的值的时候添加辅助变量记录起始位置就行了。。。
#include<iostream>
using namespace std;
int main()
{
int a[100005],dp[100005];
int t,n,i=1,maxsum,start,end,first,second;
cin>>t;
while(t--)
{
cin>>n;
for(int i=0;i<n;i++)
cin>>a[i];
dp[0]=a[0];
start=0,end=0,maxsum=-1001,first=0,second=0;
for(int i=0;i<n;i++)
{
if(dp[i-1]>=0){
dp[i]=dp[i-1]+a[i];end=i;
}
else{
dp[i]=a[i];start=end=i;
}
if(maxsum<=dp[i]){
maxsum=dp[i];
first=start;second=end;
}
}
cout<<"Case "<<i++<<":"<<endl<<maxsum<<" "<<first+1<<" "<<second+1<<endl;
if(t)
cout<<endl;
}
return 0;
}