最大m子段和(dp知识点)

讲解的好博客

 例题:

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 查看本文章

猜你喜欢

转载自blog.csdn.net/qq_38577732/article/details/89560122