『optimization 动态规划』

<更新提示>

<第一次更新>

<正文>

optimization

Description

\(visit\_world\) 发现有些优化问题可以用很平凡的技巧解决，所以他给你分享了这样一道题：

现在有一个长度为N的整数序列\(\{a_i\}\) ，你需要从中选出K个不相交的连续子区间（可以存在元素不被选），从左到右记它们的和为\(s1,s2,...,sk\)。我们的优化目标是最大化下述和式：

\[\sum_{i=1}^k|s_i−s_{i+1}|\]

你只需要输出这个最大的和即可.

Input Format

第一行两个整数N,K，意义如上.

接下来一行N个整数，第i个数表示ai，保证有\(|a_i|≤10^4\) .

Output Format

输出一行一个整数，表示答案.

Sample Input

5 3
5 2 4 3 1

Sample Output

解析

一道很神奇的\(dp\)。

关于绝对值的最大和，一个小\(trick\)就是拆开绝对值号，对正负两种情况都\(dp\)，最后去最大值一定就是最优解。

那么我们发现对于一个\(s_i\)，他可能有三种贡献系数：\(2,-2,0\)，这与\(s_{i-1},s_{i+1}\)与其的相对大小关系有关，当然，对于\(s_1,s_k\)，他们的贡献系数还有可能是\(1,-1\)，我们不妨由此设计状态。

设\(f[i][j][0/1/2/3]\)代表前\(i\)个数，分成\(j\)段，当前处于最大值(\(+2\)贡献)，最小值(\(-2\)贡献)，上升状态(\(0\)贡献)，下降状态(\(0\)贡献)的最大和。其中上升状态和下降状态指的就是最大值和最小值前的一些\(0\)贡献的状态。

然后就可以\(dp\)了，时间复杂度是\(O(n^3k)\)的。

考虑优化，第一个就是我们可以强制认为每一个数字都是要取的，当然不取可以放在\(0\)贡献的状态里处理。第二个每一段当中的数字贡献系数的相同的，可以直接一个一个添加数字。

\(Code:\)

#include <bits/stdc++.h>
using namespace std;
const int N = 3e4+20 , K = 220 , INF = 0x3f3f3f3f;
inline int read(void)
{
    int x = 0 , w = 0; char ch = ' ';
    while ( !isdigit(ch) ) w |= ch == '-' , ch = getchar();
    while ( isdigit(ch) ) x = x * 10 + ch - 48 , ch = getchar();
    return w ? -x : x;
}
int n,k,a[N],f[N][K][4];
inline void input(void)
{
    n = read() , k = read();
    for (int i=1;i<=n;i++)
        a[i] = read();
}
inline void dp(void)
{
    memset( f , 0xcf , sizeof f );
    for (int i=1;i<=n;i++)
        f[i][0][0] = f[i][0][1] = f[i][0][2] = f[i][0][3] = 0;
    f[0][0][0] = f[0][0][1] = f[0][0][2] = f[0][0][3] = 0;
    for (int i=1;i<=n;i++)
        for (int j=1;j<=min(i,k);j++)
        {
            int mul = 2;
            if ( j == 1 || j == k ) mul--;
            f[i][j][0] = max( f[i-1][j][0] , f[i-1][j-1][2] ) + mul * a[i];
            f[i][j][1] = max( f[i-1][j][1] , f[i-1][j-1][3] ) - mul * a[i];
            f[i][j][2] = max( f[i-1][j][2] , f[i][j][1] );
            f[i][j][3] = max( f[i-1][j][3] , f[i][j][0] );
            if ( mul == 1 ) continue;
            f[i][j][2] = max( f[i][j][2] , f[i-1][j-1][2] );
            f[i][j][3] = max( f[i][j][3] , f[i-1][j-1][3] );
        }
}
int main(void)
{
    input();
    dp();
    printf("%d\n",max(f[n][k][2],f[n][k][3]));
    return 0;
}

<后记>