【例题】【动规】NKOJ 3686 最大子段和

NKOJ 3686 最大子段和
时间限制 : - MS 空间限制 : 65536 KB

问题描述
给你一个包含n个整数的序列，要求从中取出m个不相交的子段，要求这m个子段的和尽肯能大，输出这个最大和。
例如，n=7,m=2, 序列如下：
-2 1 4 -2 3 -2 3
最优方案取出的两段为{1 4} {3 -2 3} 这两段的和为9

输入格式
第一行，两个整数n和m
第二行，n个空格间隔的整数，表示整数数列

输出格式
一行，一个整数，表示所求答案

样例输入
7 2
-2 1 4 -2 3 -2 3

样例输出
9

提示
1<=n <=100,000 1<=m<=300 且 m<=n
数列中的数字范围[-50000,50000]

思路：
1、阶段：从左往右讨论每一个数字。
2、状态：f[i][j]表示从前j个数中选出i段的最大和，且j必须作为最后一段结尾
3、决策：讨论第j个数加不加入第i段中
4、方程：
f[i][j]=a[1]+a[2]+a[3]+a[4]…+a[j] (i==j)
f[i][j]=max{ f[i][j-1]+a[j] , f[i-1][k]+a[j] } （i< j<=n）
//不选a[j]的情况：当讨论到a[i]后面的数时，选出的最佳方案可能不包含a[i];
边界条件：1<=i<=m, i<=j<=n, i-1<=k<=j-1

优化：
1、空间：考虑到本题空间限制64M,所以需使用滚动数组
改写方程如下：
f[i&1][j]=max{ f[i&1][j-1]+a[j] , f[(i+1)&1][k]+a[j] } （i< j<=n）
f[i&1][j]=a[1]+a[2]+a[3]+a[4]…+a[j] (i==j)

2、时间：观察到 max{f[i-1][k]}可在讨论上一循环中讨论，而不用每次都找
所以用best[j]表示选出i-1段时， 的 max{f[i-1][k]}
即，方程改为：
f[i&1][j]=a[1]+a[2]+a[3]+a[4]…+a[j] (i==j)
f[i&1][j]=max(f[i&1][j-1],best[j-1])+a[j]
边界条件：1<=i<=m, i<=j<=n, i-1<=k<=j-1

注意当讨论到f[i][j]时，所得的f[i][j]值需要更新best[j]值，但未更新的best[j]值需要在下一循环使用，所以使用temp将需要更新的best[j]值存下，等下一循环使用完毕后再更新。且当i==m循环中最终的temp值即为所求答案。

代码：

#include<cstdio> 
#include<iostream> 
using namespace std; 
#define ll long long 
const ll inf=1000000000000000000LL; 

ll a[100003]; 
ll f[2][100003],best[100003],qian[100003]; 

int main() 
{ 
    int n,m;scanf("%d%d",&n,&m); 
    for(int i=1;i<=n;i++) 
    {
        scanf("%I64d",&a[i]);
        qian[i]=qian[i-1]+a[i];//前缀和
    } 
    ll temp;
    for(int i=1;i<=m;i++) 
    { 
        temp=-inf; 
        for(int j=i;j<=n;j++) 
     {  
         if(i==j) f[i&1][j]=qian[j];//n个数中选n组必须全选
         else f[i&1][j]=max(f[i&1][j-1],best[j-1])+a[j];
         best[j-1]=temp; 
         if(f[i&1][j]>temp)
          temp=f[i&1][j]; 
     } 
    }  
    printf("%I64d",temp); 
}

第二种解决方案是只用一个变量temp存储，但注意i=1时需提前计算，以及最终答案需要单独寻找
详见代码：

#include<cstdio>  
#include<iostream>  
using namespace std;  
#define ll long long  
const ll inf=1000000000000000000LL;  

ll a[100003];  
ll f[2][100003],qian[100003];  

int main()  
{  
    int n,m;scanf("%d%d",&n,&m);  
    for(int i=1;i<=n;i++)  
    { 
        scanf("%I64d",&a[i]); 
        f[1][i]=a[i];
        qian[i]=qian[i-1]+a[i];
    }  
    ll temp; 
    for(int i=1;i<=n;i++)
    {
        if(f[1][i-1]>0) f[1][i]+=f[1][i-1];
    }
    for(int i=2;i<=m;i++)  
    {  
        temp=-inf;  
        for(int j=i;j<=n;j++)  
     {   
         if(i==j) f[i&1][j]=qian[j];
         else  
         { 
             temp=max(temp,f[(i+1)&1][j-1]); 
             f[i&1][j]=max(f[i&1][j-1],temp)+a[j]; 
         } 
     }  
    }   
    ll ans=-inf;
    for(int i=m;i<=n;i++) if(f[m&1][i]>ans) ans=f[m&1][i];//注意i范围
    printf("%I64d",ans);  
}