买卖股票的最佳时机四
之前是最多只能完成两笔交易,现在是至多可以买卖k次,那么状态数需要定为2*k+1种,此时,就要分析多种情况的递推式
找到奇偶数交替的规则即可
class Solution {
public:int maxProfit(int k, vector<int>& prices) {if(prices.size()==0) return 0;vector<vector<int>> dp(prices.size(),vector<int>(2*k+1,0));for(int j=1;j<2*k;j+=2){dp[0][j]=-prices[0];}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+1],dp[i-1][j]-prices[i]);dp[i][j+2]=max(dp[i-1][j+2],dp[i-1][j+1]+prices[i]);}}return dp[prices.size()-1][2*k];}
};
最佳买卖股票时机含冷冻期
如何去分离状态使得包含所有情况,又能更简单的得到递推式
class Solution {
public:int maxProfit(int k, vector<int>& prices) {if(prices.size()==0) return 0;vector<vector<int>> dp(prices.size(),vector<int>(2*k+1,0));for(int j=1;j<2*k;j+=2){dp[0][j]=-prices[0];}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+1],dp[i-1][j]-prices[i]);dp[i][j+2]=max(dp[i-1][j+2],dp[i-1][j+1]+prices[i]);}}return dp[prices.size()-1][2*k];}
};
最佳买卖股票时机含手续费
动态规划理解起来比贪心简单很多
只需加一个扣手续费即可
class Solution {
public:int maxProfit(vector<int>& prices, int fee) {int n=prices.size();vector<vector<int>> dp(n,vector<int>(2,0));dp[0][0]=-prices[0];for(int i=1;i<n;++i){dp[i][0]=max(dp[i-1][0],dp[i-1][1]-prices[i]);dp[i][1]=max(dp[i-1][1],dp[i-1][0]+prices[i]-fee);}return max(dp[n-1][0],dp[n-1][1]);}
};