15分钟学会 买卖股票最佳时机 中等难度问题（代码完整版）

上篇文章我们一起学习了关于买卖股票最佳时机的简单问题，相信我们都应该收获良多吧，如果还没来得及看可以看一波再来看这一篇，可能会更好理解一些吧，连接我放在下方：

https://blog.csdn.net/weixin_47468969/article/details/123246654?spm=1001.2014.3001.5501https://blog.csdn.net/weixin_47468969/article/details/123246654?spm=1001.2014.3001.5501这一次我们一起来学习最多完成K笔交易，这里需要注意的地方就是：你不能同时参与多笔交易（你必须在再次购买前出售掉之前的股票）。

这次我们假设K=2,相当于只能完成K次交易，然后分析用两次买入和两次卖出会更形象一些

首先，给数组进行初始化，分别用0代表没有操作，1代表第一次买入，2代表第一次卖出，3代表第二次买入，4代表第二次卖出。

vector<vector<int>>dp(prices.size(),vector<int>(5,0));

                                 dp[i][1]=max(dp[i-1][0]-prices[i],dp[i-1][1])

                        

                                 dp[i][2]=max(dp[i-1][1]+prices[i],dp[i-1][2])

                                 dp[i][3]=max(dp[i-1][2]-prices[i],dp[i-1][3])

                                 dp[i][4]=max(dp[i-1][3]+prices[i],dp[i-1][4])

接下来我们去初始化，通过观察我们会发现，第0天买入股票的价值就是-prices[i],所以无论是第几次买入都是dp[0][1]=dp[0][3]=-prices[i]；

最后代码就好写了，完整版coding:

#include <iostream>
#include <vector>
using namespace std;
int maxPrices(vector<int> &prices)
{
    vector<vector<int>>dp(prices.size(),vector<int>(5,0));
    dp[0][1]=-prices[0];//第一次买入
    dp[0][3]=-prices[0];//第二次买入
    for(int i=1;i<prices.size();i++)
    {
        dp[i][0]=dp[i-1][0];//什么也不操作
        dp[i][1]=max(dp[i-1][0]-prices[i],dp[i-1][1]);//第一次买入
        dp[i][2]=max(dp[i-1][1]+prices[i],dp[i-1][2]);//第一次卖出
        dp[i][3]=max(dp[i-1][2]-prices[i],dp[i-1][3]);//第二次买入
        dp[i][4]=max(dp[i-1][3]+prices[i],dp[i-1][4]);//第二次卖出
    } 
    return dp[prices.size()-1][4];
}
int main()
{
    vector<int> prices={3,3,5,0,0,3,1,4};
    int val=maxPrices(prices);
    cout<<"最大股票价值为："<<val<<endl;
    return 0;
}

刚才我们分析的是当K=2的时候出现的情况，如果K是一个未知数，我们又该如何将K带入到代码中呢

注意：当K是奇数时就是买入，当K时偶数时就是卖出

直接上代码coding

#include <iostream>
#include <vector>
using namespace std;
int maxPrices(vector<int> &prices,int k)
{
    vector<vector<int>>dp(prices.size(),vector<int>(2*k+1,0));
    for(int i=1;i<2*k;i+=2)
    {
        dp[0][i]=-prices[i];
    }
    for(int i=1;i<prices.size();i++)
    {
        for(int j=0;j<2*k-1;j+=2)
        {
            dp[i][j+1]=max(dp[i-1][j]-prices[i],dp[i-1][j+1]);
            dp[i][j+2]=max(dp[i-1][j+1]+prices[i],dp[i-1][j+2]);
        }
    }
    
    return dp[prices.size()-1][2*k];
}
int main()
{
    vector<int> prices={3,3,5,0,0,3,1,4};
    int val=maxPrices(prices,3);
    cout<<"最大股票价值为："<<val<<endl;
    return 0;
}

以上代码可以复制直接运行。

以上思想部分来源于代码随想录

接下来的代码是每买卖一次股票要添加手续费的代码

#include <iostream>
#include <vector>
using namespace std;
int maxPrices(vector<int> &price,int fee)
{
    vector<vector<int>>dp(price.size(),vector<int>(2,0));
    dp[0][0]=-price[0];
    for(int i=1;i<price.size();i++)
    {
        dp[i][0]=max(dp[i-1][0],dp[i-1][1]-price[i]);
        dp[i][1]=max(dp[i-1][1],dp[i-1][0]+price[i]-fee);//只需要在卖出的时候减去手续费即可
    }
    return dp[price.size()-1][1];

}
int main()
{
    vector<int> prices={1,3,2,8,4,9};
    int val=maxPrices(prices,2);
    cout<<"最大股票价格为："<<val<<endl;
    return 0;

}