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);
}