1057. 股票买卖 IV

给定一个长度为 N 的数组，数组中的第 i 个数字表示一个给定股票在第 ii 天的价格。

设计一个算法来计算你所能获取的最大利润，你最多可以完成 k 笔交易。

注意：你不能同时参与多笔交易（你必须在再次购买前出售掉之前的股票）。一次买入卖出合为一笔交易。

输入格式

第一行包含整数 N 和 k，表示数组的长度以及你可以完成的最大交易数量。

第二行包含 N 个不超过 1000010000 的正整数，表示完整的数组。

输出格式

输出一个整数，表示最大利润。

数据范围

1≤N≤1e5,
1≤k≤100

输入样例1：

3 2
2 4 1

输出样例1：

2

输入样例2：

6 2
3 2 6 5 0 3

输出样例2：

7

样例解释

样例1：在第 1 天 (股票价格 = 2) 的时候买入，在第 2 天 (股票价格 = 4) 的时候卖出，这笔交易所能获得利润 = 4-2 = 2 。

样例2：在第 2 天 (股票价格 = 2) 的时候买入，在第 3 天 (股票价格 = 6) 的时候卖出, 这笔交易所能获得利润 = 6-2 = 4 。随后，在第 5 天 (股票价格 = 0) 的时候买入，在第 6 天 (股票价格 = 3) 的时候卖出, 这笔交易所能获得利润 = 3-0 = 3 。共计利润 4+3 = 7.

代码：

#include<cstdio>
#include<algorithm>
#include<iostream>
#include<cstring>
using namespace std;
const int maxn=1e5+100;
int f[maxn][110][2];
int a[maxn];
int n,k;
void dp(){
    memset(f,-0x3f,sizeof(f));
    f[0][0][0]=0;
    for(int i=1;i<=n;i++)
    for(int j=0;j<=i/2;j++)
    if(!j) {
        f[i][0][0]=0;
         f[i][0][1]=max(f[i-1][0][1],f[i-1][0][0]-a[i]);
    }
    else {
        f[i][j][0]=max(f[i-1][j][0],f[i-1][j-1][1]+a[i]);
        f[i][j][1]=max(f[i-1][j][1],f[i-1][j][0]-a[i]);
    }
    return ;
}
int main(){
    cin>>n>>k;
    memset(a,0,sizeof(a));
    for(int i=1;i<=n;i++)
    scanf("%d",&a[i]);
    dp();
    int ans=-1;
    for(int i=0;i<=k;i++)
    ans=max(ans,f[n][i][0]);
    cout<<ans<<endl;
    return 0;
}

我的思路：

由于前面做的几道股票买卖都是少量的购买次数，但这道题是k次，所以自然加一维，j表示之前已经完成了几次购买，定义的东西只要能清晰表述即可不会出现与题目不同的情况即可，也可尝试一下j表示当前正在进行第几次买卖，想到了j的含义，那么就不难想到再加一维，0和1表示当前的状态，处在买入或为买入的状态，因此本题需要用三维表示，这是突破原来的第一个难点

第二个就是j的取值范围，带入几个数就会发现，j最大就是i/2的下取整，通过走两步的方法发现的

第三个需要注意的是j取零的情况，j-1必然小于零，导致数组越界，所以必须对j==0的情况单独讨论，做题的时候忽略了f[i][0][1]不是零的情况，导致调试了很长时间，所以在写代码前必须要先想好所有的步骤，再腾到代码上

第四个需要注意的点就是f[i][j][0]=max(f[i-1][j][0],f[i-1][j-1][1]+a[i])，当前状态是零的时候，若上一个状态是一，那么上一个状态的j是j-1，又是一个易错点。