DP--NYOJ子串和再续

/*
    DP
    http://acm.nyist.edu.cn/JudgeOnline/problem.php?pid=742
    题意： n 个数，求m个区间的最大和，区间大小不固定且不相交
    思路:dp[i][j] 代表前j个数的i个区间最大和
    dp[i][j] 的状态由 dp[i][j-1] 和 mx[i-1][j-1] 得到 
    mx[i-1][j-1] 代表前j-1个数分成i-1个区间中的最大值 
    dp[i][j-1] 代表前j-1个数分成i个区间的最大和 
    dp[i][j-1] + num[j] 代表第j个数合并到第i个区间
    mx[i-1][j-1] + num[j] 代表第j个数单独一组，与i-1组中最大的 
    合并成i组 
*/
# include <iostream>
# include <algorithm>
# include <string.h>
# define INFI 0x3f3f3f3f
# define MAXN 1000005
# define LL long long
using namespace std;

LL num[MAXN],dp[10][MAXN],mx[10][MAXN];
//优化 LL num[MAXN],dp[MAXN],mx[MAXN];
int main(void)
{
//  freopen("in.txt","r",stdin);
    int ncase;
    cin >> ncase;
    while(ncase--)
    {
        LL m,n;
        cin >> m >> n;
        memset(dp,0,sizeof(dp));
        memset(mx,0,sizeof(mx));
        for(int i=1; i<=n; ++i)
        {
            cin >> num[i];
        }
        LL ans;
        //未优化，内存超了 
        for(LL i=1; i<=m; ++i)
        {
            ans = -INFI;
            for(LL j=i; j<=n; ++j)
            {
                //dp[i][j] 的状态由 dp[i][j-1] 和 mx[i-1][j-1] 得到 
                dp[i][j] = max(dp[i][j-1]+num[j],mx[i-1][j-1]+num[j]);

                /*记录分成i组，前j-1个数的最大值 
                当前的ans 是上一次的dp[i][j]中的j
                也就是当前的j-1
                 */
                mx[i][j-1] = ans;
                ans = max(ans,dp[i][j]);
            }
        }
        /*
            //dp数组只用到了dp[i]，mx数组只用到了mx[i-1]
            //所以可以使用滚动数组优化 将mx和dp数组降成一维的 
            for(LL i=1; i<=m; ++i)
        {
            ans = -INFI;
            for(LL j=i; j<=n; ++j)
            {
                dp[j] = max(dp[j-1]+num[j],mx[j-1]+num[j]); 
                mx[j-1] = ans;
                ans = max(ans,dp[j]);
            }
        }
        */
        cout << ans <<endl;
    }
    return 0;
}