买卖股票问题
一、买卖股票的最佳时机
动规五部曲,确定dp数组,用二维数组记录当日持有或者不持有的状态,递推公式同样有前一天的状态决定,选择持有或者卖出,或者不做任何动作
class Solution {
public:
int maxProfit(vector<int>& prices) {
//动规五部曲
//贪心
vector<vector<int>>dp(prices.size(), vector<int>(2));
//dp[i][0]表示第i天持有股票得到的最多现金,dp[i][1]表示第i天不持有股票得到的最多现金
dp[0][0] = -prices[0];
dp[0][1] = 0;
//if (prices.size() == 0) return 0;
for (int i = 1; i < prices.size(); i++) {
dp[i][0] = max (dp[i-1][0], -prices[i]);
dp[i][1] = max (dp[i-1][1], dp[i-1][0] + prices[i]);
}
return dp[prices.size()-1][1];
}
};
二、买卖股票的最佳时机II
与I不同的是,Ⅱ可以买卖多次,所以需要更改当日持有状态下买入操作
class Solution {
public:
int maxProfit(vector<int>& prices) {
//和I的差别在于可以买多次
if (prices.size() == 0) return 0;
vector<vector<int>>dp(prices.size(), vector<int>(2));
dp[0][0] = -prices[0];
dp[0][1] = 0;
for (int i = 1; i < prices.size(); 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]);
}
return dp[prices.size()-1][1];
}
};
总结
dp数组有点想不出来
学习时间90min。
学习资料:《代码随想录》。