AcWing 1057. 股票买卖 IV

#include<bits/stdc++.h>
using namespace std;

const int N = 1e5+110,K = 110; 
int f[N][K][2];
//f[i][0/1]表示前i天，共完成j次交易时获得的最大利润。
//0表示当前未持有股票，1表示当前持有股票
int w[N];
int n,k;

int main()
{
    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;i++)
    scanf("%d",&w[i]);
    
    memset(f,0xcf,sizeof f);
    for(int i=0;i<=n;i++)f[i][0][0]=0;
    
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=k;j++)
        {
            //第一支股票如果买入时，必须按照一次交易处理。
            //否则如果股票如果买入时，不按一次交易处理，代表着股票卖出才算一次交易，
            //也就代表着在第一支股票卖出之前还买了一支股票，显然是矛盾的。
            f[i][j][0]=max(f[i-1][j][1]+w[i],f[i-1][j][0]);
            f[i][j][1]=max(f[i-1][j-1][0]-w[i],f[i-1][j][1]);
        }
    }
    int res=0;
    for(int i=0;i<=k;i++)
    res=max(res,f[n][i][0]);
    cout<<res;
    return 0;
}