http://acm.hdu.edu.cn/showproblem.php?pid=1024
题意:输入两个数m, n然后输入 n 个数,求将这n个数分出 m 个不相交的子序列,求所有子序列的和
思路:看完题目应该就知道这是个dp,dp[i][k], 表示将序列前 k 前项分成 i 段而且arg[k] 在序列里的最大和。但是题目给出的 n 可以到1000000,然而 m 没说有多大(不过我估计也不大,要不然O(n*m)很容易超时),不管 m 多大,就算m只有 10 也超内存了吧。先看看转移方程吧,说不定就找到缩小内存的方法呢。
dp[i][k] = max( dp[i-1][1…..k-1], dp[i][k-1] ) +arg[k] ;
很容易看出我们的 dp[i][k] 只由 dp[i-1][k] 和 dp[i][k-1] 传递。所以想到用滚动数组转移方程就是;
dp[i&1][k] = max( dp[(i-1)&1][1…..k-1], dp[i&1][k-1] ) +arg[k] ;
#include <cstdio>
#include <cstring>
#include <iostream>
using namespace std;
typedef long long loli;
const loli inf = loli(1) << 60;
int n, m;
int arg[1000010];
loli dp[2][1000010];
int main()
{
ios::sync_with_stdio(false);
while(cin>>m>>n)
{
for(int i = 1; i <= n; i++){
cin>>arg[i];
}
memset(dp, 0, sizeof(dp));
for(int i = 1; i <= m; i++){
loli maxn =dp[(i-1)&1][i-1];
dp[i&1][i] = dp[(i-1)&1][i-1] + arg[i]; // i 个数 分成 i 段,必须全部加上
for(int k = i+1; k <= n; k++){
if(maxn < dp[(i-1)&1][k-1]) //保存dp[(i-1)&1][1…..k-1]中的最大值,减少一层循环
maxn = dp[(i-1)&1][k-1];
dp[i&1][k] = max( maxn + arg[k], dp[i&1][k-1] + arg[k]);
}
}
loli res = -inf;
for(int i = m; i <= n; i++){
if(res < dp[m&1][i])
res = dp[m&1][i];
}
cout<<res<<endl;
}
return 0;
}