描述:
给一些列数字,表示每条股票的价格,如果可以买卖一次(不能同一天买和卖),求最大利益(即差最大)。
其他三道问题是,如果能买卖无限次,买卖两次,买卖k次。
题一:
实质是求后面一个数减前一个数的最大差值。
维护一个最小值,和当前最大值。只需遍历一次,空间也是常数。
int maxProfit(vector<int>& prices) {if (prices.size() < 1)return 0;int min_ = prices[0];int ret = 0;for (int i = 1; i < prices.size(); i++) {ret = max(ret, prices[i] - min_);min_ = min(min_, prices[i]);}return ret; }
题二:
只要是后一个数比前一个大,都增。
int maxProfit(vector<int>& prices) {if (prices.size() < 1)return 0;int ret = 0;for (int i = 0; i < prices.size() - 1; i++) {ret += max(prices[i + 1] - prices[i], 0);}return ret; }
题三:
可进行两次操作。
其中一个思路,可以关注分界点,可以枚举分界点,求左右两边的最优操作,在LeetCode会超时,显然,复杂度n^2。
思考下优化,我们可以计算每个点的最大值,左边不用重复计算,每次分界点往左移,都像题一那样计算最大值即可;
而右边,其实可以反向计算一遍,但是,右边改成求最小值。
最后加起来即可。
int maxProfit(vector<int>& prices) {int size = prices.size();if (size < 1)return 0;int* left = new int[size]{0};int* right = new int[size]{0};int ret = 0;int lmin = prices[0];int lmax = 0;for (int i = 1; i < size; i++) {lmax = max(lmax, prices[i] - lmin);left[i] = lmax;lmin = min(lmin, prices[i]);}int rmin = 0;int rmax = prices[size - 1];for (int i = size - 1; i >= 0; i--) {rmin = min(rmin, prices[i] - rmax);right[i] = -rmin;rmax = max(rmax, prices[i]);}for (int i = 0; i < size - 1; i++) {ret = max(ret, left[i] + right[i + 1]);}return max(ret, left[size - 1]); }
思路二:
int maxProfit(vector<int>& prices) {int n = prices.size();if(n==0) return 0;int sell1 = 0, sell2 = 0, buy1 = INT_MIN, buy2 = INT_MIN;for(int i =0; i<n; i++){buy1 = max(buy1, -prices[i]);sell1 = max(sell1, prices[i]+buy1);buy2 = max(buy2, sell1-prices[i]);sell2 = max(sell2, prices[i]+buy2);} return sell2; }
题四:
动态规划:
其中diff表示今天和昨天的差。
global[i][j] = max(local[i][j], global[i-1][j])
local[i][j] = max(global[i-1][j-1] + max(diff,0), local[i-1][j] + diff)
local[i][j]表示最后一次卖出在今天的最大利益,局部最优。
global[i][j]表示全局最优。
第一条式子:要么在今天卖出最优,要么前一天的全局最优。
第二条式子:前者为之前的全局最优加上最后一次交易在今天。
注意diff,我们要的是不大于j的交易次数;
如果i - 1天还持有,则i天卖出,共j - 1次操作;如果i-1天不持有,则i - 1天买入,i天卖出,共j次操作。
后者为i - 1天卖出加上今天diff,表示i - 1天还持有,加上今天的。
int maxProfit(int k, vector<int>& prices) { if (prices.size() < 2) return 0; int days = prices.size(); if (k >= days) return maxProfit2(prices);auto local = vector<vector<int> >(days, vector<int>(k + 1));auto global = vector<vector<int> >(days, vector<int>(k + 1));for (int i = 1; i < days ; i++) {int diff = prices[i] - prices[i - 1]; for (int j = 1; j <= k; j++) { local[i][j] = max(global[i - 1][j - 1], local[i - 1][j] + diff); global[i][j] = max(global[i - 1][j], local[i][j]); } } return global[days - 1][k]; } int maxProfit2(vector<int>& prices) { int maxProfit = 0; for (int i = 1; i < prices.size(); i++) { if (prices[i] > prices[i - 1]) { maxProfit += prices[i] - prices[i - 1]; } } return maxProfit; }
类似题三的做法:
int maxProfit(int k, vector<int>& prices) {int n = prices.size();if(k>n/2){int buy = INT_MIN, sell = 0;for(int i=0; i<n; i++){buy = max(buy, sell-prices[i]);sell = max(sell, buy+prices[i]);}return sell;}vector<int> sell(k+1, 0);vector<int> buy(k+1, 0);for(int i=0; i<=k; i++) buy[i] = INT_MIN;for(int i=0; i<n; i++){for(int j=1; j<k+1; j++){buy[j] = max(buy[j], sell[j-1]-prices[i]);sell[j] = max(sell[j], buy[j]+prices[i]);}}return sell[k]; }