力扣第40天----第121题、第122题
文章目录
一、第121题--买卖股票的最佳时机
分2种情况考虑,根据持有股票、不持有股票这2种情况,完成递推公式。另外,这里要求只买卖一次,所以买入的前一个状态一定是手里现金为0。
c++
class Solution {
public:
int maxProfit(vector<int>& prices) { //0表示不持有, 1表示持有
vector<vector<int>> dp(2, vector<int>(prices.size(), 0));
dp[0][0] = 0;
dp[1][0] = -prices[0];
for(int i =1; i<prices.size(); ++i){
dp[0][i] = max(dp[0][i-1], dp[1][i-1] + prices[i]);
dp[1][i] = max( - prices[i], dp[1][i-1] ); //这里控制,只买卖一次。如果是别的值(不是0),就是买卖多次。
}
return dp[0][prices.size() - 1];
}
};
二、第122题--买卖股票的最佳时机II
跟上一题,基本一样。可以买卖多次,那么买入的前一个状态,手里可以有现金,要根据前一个状态手里现金数,再做处理。
c++
class Solution {
public:
int maxProfit(vector<int>& prices) { //0表示不持有, 1表示持有
vector<vector<int>> dp(2, vector<int>(prices.size(), 0));
dp[0][0] = 0;
dp[1][0] = -prices[0];
for(int i =1; i<prices.size(); ++i){
dp[0][i] = max(dp[0][i-1], dp[1][i-1] + prices[i]);
dp[1][i] = max(dp[0][i-1] - prices[i], dp[1][i-1] );
}
return dp[0][prices.size() - 1];
}
};