例题:
51nod1052
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
using namespace std;
typedef long long ll;
const int maxn = 5010;
const int INF = 1e9;
//ll dp[maxn][maxn];
ll dp[2][maxn];
ll a[maxn];
int n, m;
int main()
{
cin >> n >> m;
for(int i = 1; i <= n; i++)
cin >> a[i];
for(int i = 0; i <= n; i++)
{
dp[0][i] = 0, dp[1][i] = 0;
}
for(int i = 1, k = 1; i <= m; i++, k ^= 1)
{
ll maxx = -INF; //maxx为用于记录dp[i - 1][i -1 ~ j - 1]的最大值。因为对于j == i时:dp[i][i](注释的dp数组为没有优化的形式,即dp[m][n],便于理解)必然需要接在dp[i-1][i-1]后面,此处初始化为-INF是为了下面for循环第二行j == i时maxx必然被dp[i-1][i-1]更新
for(int j = i; j <= n; j++) //讲解博客优化为n - m + i,个人习惯到n
{
dp[k][j] = max(dp[k][j], dp[k][j - 1] + a[j]);
maxx = max(maxx, dp[k ^ 1][j - 1]);
dp[k][j] = max(dp[k][j], maxx + a[j]);
}
}
ll ans = -(ll)maxn * INF; //WA点之一,注意初始化ans根据题意。如可以不必须选m段,要求ans>=0(此题要求正数同理), 则初始化为0;如果必须选够m段,则初始化为数据个数*数据最小值。
for(int i = m; i <= n; i++)
ans = max(ans, dp[m&1][i]); //m&1非0即1, 根据m和k的规律可以这样用,巧妙
cout << ans << endl;
return 0;
}
扫描二维码关注公众号,回复:
6091791 查看本文章