Best time to buy and sell stocks
典型的动态规划问题,给了一个数组,其中记录了每一天的股票值,可以买也可以卖,要先买才能卖,在卖的时候有交易费,返回最大获利的值。对于每一天的,只有两个选择,买和卖。如果买的话,因为是先买才能卖,所以用前一天卖获利的最大值减去当日股票价格,与前一天买的获利值作比较,取最大。如果卖的话,同样是先买了才能卖,所以用前一天买的最大利润减去今日价格再减去交易费用,与前一天卖的获利值比较,取最大值。只需要遍历一遍就可以,所以时间复杂度是O(n)。
以下是代码:
class Solution {
public:
int maxProfit(vector<int>& prices, int fee) {
int buy[prices.size() + 1] = {0}, sell[prices.size() + 1] = {0};
buy[0] = -prices[0];
for(int i = 1; i < prices.size(); i++) {
buy[i] = max(buy[i - 1], sell[i - 1] - prices[i]);
sell[i] = max(sell[i - 1], buy[i - 1] + prices[i] - fee);
}
return sell[prices.size() - 1];
}
};