123.买卖股票的最佳时机III
题目链接:123
题目描述:
给定一个数组,它的第 i 个元素是一支给定的股票在第 i 天的价格。
设计一个算法来计算你所能获取的最大利润。你最多可以完成 两笔 交易。
注意:你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。
/*
1. 0表示不操作,1表示第一次持有,2表示第一次不持有,3表示第二次持有,4表示第二次不持有
**/
class Solution {
public:
int maxProfit(vector<int>& prices) {
vector<vector<int>>dp(prices.size()+1,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][1],dp[i-1][0]-prices[i]);
dp[i][2]=max(dp[i-1][1]+prices[i],dp[i-1][2]);
dp[i][3]=max(dp[i-1][3],dp[i-1][2]-prices[i]);
dp[i][4]=max(dp[i-1][3]+prices[i],dp[i-1][4]);
}
return dp[prices.size()-1][4];
}
};
188.买卖股票的最佳时机IV
题目链接:188
题目描述:
给定一个整数数组 prices ,它的第 i 个元素 prices[i] 是一支给定的股票在第 i 天的价格,和一个整型 k 。
设计一个算法来计算你所能获取的最大利润。你最多可以完成 k 笔交易。也就是说,你最多可以买 k 次,卖 k 次。
注意:你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。
/**
跟上一题一样,dp[i][0]表示无操作,1,3,5分别表示第k次持有,而2,4,6表示不持有。
对于持有,则dp[i][k]=-prices[i];
*/
class Solution {
public:
int maxProfit(int k, vector<int>& prices) {
vector<vector<int>>dp(prices.size(),vector<int>(2*k+1,0));
if(prices.size()==0) return 0;
for(int i=1;i<2*k;i+=2)
dp[0][i]=-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];
}
};