【51nod】1052 最大M子段和 DP

题目传送门

好吧，好像是我ZZ了，原来连续的一段数可以当做许多段数来看的……

（突然掀桌，气得我都变成简笔画了！害得我WA了6发！）

其实这题就是一道非常水的DP题，定义$f[i][j][0/1]$表示前$i$个数选了$j$段，第三维表示当前的数是否被选在第$j$段数内。然后直接进行状态转移就行了。

突然发现这样开数组会$MLE$，又发现好像第$i$个数的状态只和第$i-1$个数的状态有关，那么我们直接用滚动数组来优化内存就行了。

附上AC代码：

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

const int N=5010;
long long n,m,a[N],f[2][N][2],sum,cnt;

int main(void){
    scanf("%lld%lld",&n,&m),memset(f,0x80,sizeof f);
    for (int i=1; i<=n; ++i) scanf("%lld",&a[i]),a[i]>0?sum+=a[i],++cnt:0;
    if (m>=cnt){printf("%lld\n",sum);return 0;}
    f[0][0][0]=0;
    for (int i=1; i<=n; ++i){
        int now=i&1;
        memset(f[now],0x8f,sizeof f[now]),f[now][0][0]=0;
        for (int j=1; j<=min(m,1ll*(i+1)/2); ++j){
            f[now][j][0]=max(f[now^1][j][0],f[now^1][j][1]);
            f[now][j][1]=max(max(f[now^1][j][1],f[now^1][j-1][0]),f[now^1][j-1][1])+a[i];
        }
    }
    return printf("%lld\n",max(f[n&1][m][0],f[n&1][m][1])),0;
}

以上是本蒟蒻的辣鸡DP，看了其他大佬的博客后发现自己的想法就像一个ZZ……

设外层循环$m$次，第$k$次循环表示把整个序列分成$k$个子段所能得到的最大子段和。

定义$pre$数组表示把整个序列分成$k-1$个子段所能得到的最大子段和。

显然动规数组$f[i]$有两种转移方式：

1. 前$i-1$个数分成了$k$个子段，把当前的数加到第$k$个子段的末尾。
2. 前$i-1$个数分成了$k-1$个子段，把当前的数作为第 k <script type="math/tex" id="MathJax-Element-19">k</script>个子段的开头。

这样的DP显然比我的想法巧妙了许多，省了许多内存，代码量也小了许多。

要尽快改进自己的思维方式，不要变成一个数据结构学傻掉的人。

附上AC代码：

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

typedef long long ll;
const int N=5010;
int n,m,a[N];
ll pre[N],f[N];

int main(void){
    scanf("%d%d",&n,&m);
    for (int i=1; i<=n; ++i) scanf("%d",&a[i]);
    memset(f,0x80,sizeof f);
    while (m--){
        ll ans=-1ll<<60;
        for (int i=1; i<=n; ++i){
            if (i==1) f[i]=pre[i-1]+a[i];
            else f[i]=max(f[i-1],pre[i-1])+a[i];
            pre[i-1]=ans,ans=max(ans,f[i]);
        }
        pre[n]=ans;
    }
    return printf("%d\n",pre[n]),0;
}