题意:
长度为n的数列,选择m个不想交的区间,所有区间数字和最大。
题解:
题目上没说m的范围,我还想着这怎么写,后来发现AC代码都是O(nm)的
首先想一个最简单的DP, d p [ i ] [ j ] dp[i][j] dp[i][j]代表是前 j j j个数字分成 i i i个区间的最大区间和。那么状态转移方程为
- 第j个数不是新的区间。 d p [ i ] [ j ] = d p [ i ] [ j − 1 ] + a [ j ] dp[i][j]=dp[i][j-1]+a[j] dp[i][j]=dp[i][j−1]+a[j]
- 第j个数是新的区间。 d p [ i ] [ j ] = m a x i < k < j ( d p [ i − 1 ] [ k ] ) + a [ j ] dp[i][j]=max_{i<k<j}(dp[i-1][k])+a[j] dp[i][j]=maxi<k<j(dp[i−1][k])+a[j]
不是新区间,那么一定和 j − 1 j-1 j−1连在一起,否则一定就是新的区间了。
考虑到,每个 i i i只跟 i − 1 i-1 i−1和 i i i有关,可以优化转换成一维数组。
外层循环设置成 i i i递增,这样保证 i i i与 i − 1 i-1 i−1的转移不会错;内层循环设置 j j j递增,这样 j j j与 j − 1 j-1 j−1的转移得到保证。
用 p r e [ j − 1 ] pre[j-1] pre[j−1]表示前 j − 1 j-1 j−1个的最大和。
AC代码:
#include <cstdio>
#include <iostream>
#include <vector>
#include <string>
#include <queue>
#include <algorithm>
#include <cmath>
#include <set>
#include <map>
#include <iomanip>
#include <cstdlib>
#include <cstring>
#define rep(i,a,b) for(int i=(a);i<=(b);i++)
#define lep(i,a,b) for(int i=(a);i>=(b);i--)
#define pii pair<int,int>
#define pll pair<long long,long long>
#define mp make_pair
#define pb push_back
#define fir first
#define sec second
#define All(x) x.begin(),x.end()
#define ms(a,b) memset(a,b,sizeof(a))
#define INF 0x3f3f3f3f
#define INFF 0x3f3f3f3f3f3f3f3f
#define multi int T;scanf("%d",&T);while(T--)
using namespace std;
typedef long long ll;
typedef double db;
const int N=1e6+5;
const int mod=10007;
const db eps=1e-6;
const db pi=acos(-1.0);
int n,m,s[N];
ll pre[N],dp[N],tmp;
int main()
{
#ifndef ONLINE_JUDGE
freopen("D:\\work\\data.in","r",stdin);
#endif
while(~scanf("%d%d",&m,&n)){
rep(i,1,n){
scanf("%d",&s[i]);
pre[i]=dp[i]=0;
}
rep(i,1,m){//分成i+1块只与i块有关,所以可以省去一重空间
tmp=-INF;
rep(j,i,n){
dp[j]=max(dp[j-1],pre[j-1])+s[j];
pre[j-1]=tmp;//不包括自己的最大的前面的
tmp=max(tmp,dp[j]);
}
}
printf("%lld\n",tmp);
}
}