首页 > 代码库 > POJ2479,2593: 两段maximum-subarray问题

POJ2479,2593: 两段maximum-subarray问题

虽然是两个水题,但是一次AC的感觉真心不错

这个问题算是maximum-subarray问题的升级版,不过主要算法思想不变:

1. maximum-subarray问题

maximum-subarray就是找到数组A[1....n]中的连续子数组A[i.....j]并且A[i]+...+A[j]和最大。当然了,(1<=i<=j<=n)。

maximum-subarray的O(n)解法就是从左到右扫描数组A,另外设置一工具数组DP,DP数组的作用就是记录以当前下标为终止下标的子数组和。比如DP[j]=A[i]+....+A[j](A[i] 到 A[j]是连续的)。现在假设我们已经求出DP[j],数组即将扫描A[j+1],则DP[j]与DP[j+1]的关系描述如下:

DP[j+1]=(DP[j]>0)? (DP[j]+A[j+1]) : (A[j+1]).

因为DP[j+1]要求出以j+1下标为结束下标的子数组和,而且DP[j]已经求出,所以我们要判断DP[j]是否为正数,如为正,则加上A[j+1]。如为负,那么很明显的,A[i]+.....+A[j]+A[j+1]<A[j+1], 所以要让DP[j+1]=A[j+1]。

2. 求两个maximum-subarray问题

这两个maximum-subarray不相交。

技术分享

 

我们可以设置一个“分水岭”,假设为k,那么maximum-subarray(A[1..k]) + maximum-subarray(A[k+1..n])就是我们要的解。

当然如果我们枚举每一个k值(1<=k<=n-1)的话,因为题目开出的N值为50000,真个时间复杂度为O(n^2),必然超时。

所以我们可以再设两个工具数组:lmax和rmax,lmax[i]表示A[1]到A[i]的最大子数组和,rmax[i]=A[i+1]....A[n]的最大子数组和。再次明确一下:lmax/rmax数组与DP数组的不同。

假设DP[s]=A[p+..q+..+r+..s], 那么lmax[s]可能就等于A[q+...+r]或者A[p+...+r]或者等等。

然后我们得到每个lmax[k]+rmax[k],对k进行枚举,lmax[k]+rmax[k]值最大的即为最后的解。

附上POJ2593代码:(POJ2479改动一点就可以了)

#include<iostream>#include<algorithm>#include<cstdio>#include<cstring>#include<map>#include<vector>using namespace std;const int max_size=100010;int n,a[max_size];int ldp[max_size],rdp[max_size],ans,inf=1<<30;int lmax[max_size],rmax[max_size];int main(){    while(scanf("%d",&n)!=EOF&&n){        memset(a,0,sizeof(a));        memset(ldp,0,sizeof(ldp));        memset(rdp,0,sizeof(rdp));        memset(lmax,0,sizeof(lmax));        memset(rmax,0,sizeof(rmax));        ans=-inf;        for(int i=1;i<=n;i++){            scanf("%d",&a[i]);        }         lmax[1]=ldp[1]=a[1];         for(int i=2;i<=n;i++){            if(ldp[i-1]>0) ldp[i]=ldp[i-1]+a[i];            else ldp[i]=a[i];            ans=max(ans,ldp[i]);            lmax[i]=ans;        }        rmax[n]=rdp[n]=a[n];        ans=-inf;        for(int i=n-1;i>=1;i--){            if(rdp[i+1]>0) rdp[i]=rdp[i+1]+a[i];            else rdp[i]=a[i];            ans=max(ans,rdp[i]);            rmax[i]=ans;        }        ans=-inf;        for(int k=1;k<=n-1;k++){            ans=max(ans,lmax[k]+rmax[k+1]);        }        printf("%d\n",ans);    }}

 

POJ2479,2593: 两段maximum-subarray问题