买卖股票的最佳时机I
动态规划开始不只是一维的了,升级成了二维
第二个维度只有两个,一个是持有股票,一个是不持有股票
dp[i][0]代表第i天持有股票的情况下,最大的收益。
dp[i][1]代表第i天不持有股票的情况下,最大的收益。
持有股票有两种可能,一是前一天已经持有了dp[i-1][0],二是前一天没有持有,今天购入股票(即当天买的,由于只能买一次,因此这是第一次买,所以最大价值就是0 - prices[i]),选最大值就可以。
不持有股票也有两种可能,一是前一天就没有持股票dp[i - 1][1], 二是前一天持股票了,今天卖了(即dp[i-1][0]+prices[i])
int maxProfit(vector<int>& prices) {
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(-prices[i], dp[i - 1][0]);
dp[i][1] = max(dp[i - 1][0] + prices[i], dp[i - 1][1]);
}
return dp[prices.size() - 1][1];
}
买卖股票的最佳时机II
与上一题不同的点在于,可以购买多次
持有股票有两种可能,一是前一天已经持有了dp[i-1][0],二是前一天没有持有,今天购入股票(即当天买的,由于可以买多次,并非第一次买,所以手里的钱不一定是0,而是dp[i -1][1], 所以最大价值就是dp[i -1][1] - prices[i]), 选最大值就可以。
不持有股票也有两种可能,一是前一天就没有持股票dp[i - 1][1], 二是前一天持股票了,今天卖了(即dp[i-1][0]+prices[i])
int maxProfit(vector<int>& prices) {
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][1] - prices[i], dp[i - 1][0]);
dp[i][1] = max(dp[i - 1][0] + prices[i], dp[i - 1][1]);
}
return max(dp[prices.size() - 1][1], dp[prices.size() - 1][1]);
}