股票买卖最大收益总结-Best Time to Buy and Sell Stock

    1. Best Time to Buy and Sell Stock III

Say you have an array for which the ith element is the price of a given stock on day i.
Design an algorithm to find the maximum profit. You may complete at most two transactions.
Note: You may not engage in multiple transactions at the same time (i.e., you must sell the stock before you buy again).

分析题目的意思就是说可以买卖一次或者两次甚至不买卖,求出最大化的收益。
如果只买卖一次那么问题就变成另一个问题:只买卖一次能够获得的最大收益。这个问题采用动归的递推形式就能够解决。而买卖两次的情况是可以化成买卖1次的最大值问题。

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        int n = prices.size();
        if(n<=1) return 0;
        
        vector<int> f(n,0),g(n,0);
        int temp = INT_MAX,cur_max = 0;
        //f[i]为前i个股票最多只买卖一次能够获得最大收益
        for(int i=0;i<n;++i) {
            temp = min(temp,prices[i]);
            f[i] = max(cur_max,prices[i]-temp);
            cur_max = f[i];
        }
        temp = 0;
        cur_max = 0;
        //g[i]为后n-i个股票最多只买卖一次能够获得的最大收益
        for(int i=n-1;i>=0;--i) {
            temp = max(temp,prices[i]);
            g[i] = max(cur_max,temp - prices[i]);
            cur_max = g[i];
        }
        //总的收益是所有可能选择的2分段的最大值
        int res = 0;
        for(int i=0;i<n;++i) {
            res = max(res,g[i]+f[i]);
        }
        return res;
    }
};

这不是一道典型的动归,但是也是使用了动态规划的思想。
首先很容易想到将问题拆分成两段,每一段对应一个只用最多一次交易获取的最大值。这样只需要将两段加起来就是整体的最大值。对于划分成两段的划分方法有n种,所以只需要遍历这n种划分选择,求最大值即可。
下面就是怎么求一段序列prices[0:i]当中只最多买卖一次可以获得的最大值。这个问题稍微思考一下就知道,只需要遍历整个序列,每次更新一个最小值变量temp保存当前遇到的最小值,那么当前可以获得的最大收益为
g[i] = max(prices[k] - temp, g[i-1])
当前的最优解只和前一个阶段的最优解有关,这不就是DP的最优子结构么~,由于当前最优解只和前一个阶段的最优解有关,不需要做出选择,所以就是简单的递推形式。如果从自顶向下的形式去分析就不行了,因为
dp(i) = max(dp(i-1), prices[i] - m[i-1]) ,m[i-1]为0…i-1区间的最小值。
m[i] = min(prices[i], m[i-1])
也就是说自顶向下还得开一个数组去记录每一个阶段的最小值:
写出来的形式就是:

def dp(i,prices,m):
	if i == 0:
		m[i] = prices[0]
		return 0
		
	 cur = max(dp(i-1,prices,m),prices[i] - m[i-1])
	 m[i] = min(prices[i], m[i-1]) 
	 return cur

所以这样的问题还是自底向上写比较的顺手,但是就是容易忽视其DP的本质。

做这样的题目还是不能先上来当做直接的dp去做,需要先从题目本身上对题目进行相应的抽象,转化成容易解决的子问题,对问题进行转化,再对抽象的问题进行模型的匹配求解。一个问题必然需要考察某一个知识点,所以扒开问题的外衣找到真实的考点是关键的第一步。然后才是分析问题采用最合适的算法去解决。

class Solution {
public:
    int maxProfit(vector<int> &prices) {
        // f[k, ii] represents the max profit up until prices[ii] (Note: NOT ending with prices[ii]) using at most k transactions. 
        // f[k, ii] = max(f[k, ii-1], prices[ii] - prices[jj] + f[k-1, jj]) { jj in range of [0, ii-1] }
        // = max(f[k, ii-1], prices[ii] + max(f[k-1, jj] - prices[jj]))
        // f[0, ii] = 0; 0 times transation makes 0 profit
        // f[k, 0] = 0; if there is only one price data point you can't make any money no matter how many times you can trade
        if (prices.size() <= 1) return 0;
        else {
            int K = 2; // number of max transation allowed
            int maxProf = 0;
            vector<vector<int>> f(K+1, vector<int>(prices.size(), 0));
            for (int kk = 1; kk <= K; kk++) {
                int tmpMax = f[kk-1][0] - prices[0];
                for (int ii = 1; ii < prices.size(); ii++) {
                    f[kk][ii] = max(f[kk][ii-1], prices[ii] + tmpMax);
                    tmpMax = max(tmpMax, f[kk-1][ii] - prices[ii]);
                    maxProf = max(f[kk][ii], maxProf);
                }
            }
            return maxProf;
        }
    }
};

全部评论

相关推荐

牛客5655:其他公司的面试(事)吗
点赞 评论 收藏
分享
4 4 评论
分享
牛客网
牛客企业服务