买卖股票的最好时机(三)_牛客题霸_牛客网 (nowcoder.com)
BM82 买卖股票的最好时机(三)
描述
假设你有一个数组prices,长度为n,其中prices[i]是某只股票在第i天的价格,请根据这个价格数组,返回买卖股票能获得的最大收益
1. 你最多可以对该股票有两笔交易操作,一笔交易代表着一次买入与一次卖出,但是再次购买前必须卖出之前的股票
2. 如果不能获取收益,请返回0
3. 假设买入卖出均无手续费
数据范围:1 \le n \le 10^51≤n≤105,股票的价格满足 1 \le val\le 10^41≤val≤104
要求: 空间复杂度 O(n),时间复杂度 O(n)
进阶:空间复杂度 O(1),时间复杂度 O(n)
示例1
输入:[8,9,3,5,1,3]
返回值:4
说明:第三天(股票价格=3)买进,第四天(股票价格=5)卖出,收益为2 第五天(股票价格=1)买进,第六天(股票价格=3)卖出,收益为2 总收益为4。
示例3
输入:[1,2,8,3,8]
返回值:12
说明:第一笔股票交易在第一天买进,第三天卖出;第二笔股票交易在第四天买进,第五天卖出;总收益为12。 因最多只可以同时持有一只股票,所以不能在第一天进行第一笔股票交易的买进操作,又在第二天进行第二笔股票交易的买进操作(此时第一笔股票交易还没卖出),最后两笔股票交易同时在第三天卖出,也即以上操作不满足题目要求。
思路:对于此类求极值且情况多,前面的选择会影响后面的结果和选择时,考虑使用贪心和动态规划
初始状态:f[0][0]=0(第一天不买)
f[0][3]=-prices[0](第一天购买)
递推方式:
class Solution {
public:
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
* 两次交易所能获得的最大收益
* @param prices int整型vector 股票每一天的价格
* @return int整型
*/
int maxProfit(vector<int>& prices) {
// write code here
vector<vector<int>> f(prices.size(), vector<int>(5, -10000));
f[0][0] = 0;
f[0][3] = -1 * prices[0];
for (int i = 1; i < prices.size(); i++) {
f[i][0] = f[i-1][0];
f[i][1] = max(f[i - 1][1], f[i - 1][3] + prices[i]);
f[i][2] = max(f[i - 1][2], f[i - 1][4] + prices[i]);
f[i][3] = max(f[i - 1][3], f[i - 1][0] - prices[i]);
f[i][4] = max(f[i - 1][4], f[i - 1][1] - prices[i]);
}
//最后不持有股票
return max(max(0,f[prices.size() - 1][2]), f[prices.size() - 1][1]);
}
};