首页 > 其他 > 详细

121. 122. 123. 188. Best Time to Buy and Sell Stock *HARD* -- 买卖股票

时间:2016-05-04 18:35:21      阅读:226      评论:0      收藏:0      [点我收藏+]

121.

Say you have an array for which the ith element is the price of a given stock on day i.

If you were only permitted to complete at most one transaction (ie, buy one and sell one share of the stock), design an algorithm to find the maximum profit.

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        int n = prices.size();
        if(n < 1)
            return 0;
        int mini = prices[0], ans = 0, i;
        for(i = 1; i < n; i++)
        {
            if(prices[i]-mini > ans)
                ans = prices[i]-mini;
            if(prices[i] < mini)
                mini = prices[i];
        }
        return ans;
    }
};

 

122.

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 as many transactions as you like (ie, buy one and sell one share of the stock multiple times). However, you may not engage in multiple transactions at the same time (ie, you must sell the stock before you buy again).

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        int n = prices.size();
        if(n < 1)
            return 0;
        int mini = prices[0], maxi = prices[0], ans = 0, i;
        for(i = 1; i < n; i++)
        {
            if(prices[i] > maxi)
                maxi = prices[i];
            else if(prices[i] < maxi)
            {
                ans += maxi - mini;
                maxi = mini = prices[i];
            }
        }
        ans += maxi - mini;
        return ans;
    }
};

 

123.

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.

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        int n = prices.size();
        if(n < 1)
            return 0;
        vector<int> forward(n, 0), backward(n, 0);
        int mini, maxi, ans, i;
        forward[0] = 0;
        mini = prices[0];
        for(i = 1; i < n; i++)
        {
            forward[i] = max(forward[i-1], prices[i] - mini);
            if(prices[i] < mini)
                mini = prices[i];
        }
        backward[n-1] = 0;
        maxi = prices[n-1];
        for(i = n-2; i >= 0; i--)
        {
            backward[i] = max(backward[i+1], maxi - prices[i]);
            if(prices[i] > maxi)
                maxi = prices[i];
        }
        ans = 0;
        for(i = 0; i < n; i++)
        {
            ans = max(ans, forward[i] + backward[i]);
        }
        return ans;
    }
};

 

188.

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 k transactions.

Note:
You may not engage in multiple transactions at the same time (ie, you must sell the stock before you buy again).

class Solution {
public:
    int maxProfit(int k, vector<int>& prices) {
        int n = prices.size(), i, j;
        if(n < 1)
            return 0;
        if(k >= (n>>1))
        {
            int ans = 0;
            for(i = 0; i < n-1; i++)
            {
                if(prices[i+1]-prices[i] > 0)
                    ans += prices[i+1]-prices[i];
            }
            return ans;
        }
        vector<int> buy(k+1, INT_MIN), sell(k+1, 0);
        for(i = 0; i < n; i++)
        {
            for(j = 1; j <= k; j++)
            {
                buy[j] = max(buy[j], sell[j-1] - prices[i]);
                sell[j] = max(sell[j], buy[j] + prices[i]);
            }
        }
        return sell[k];
    }
};

buy[i]表示买i个最多剩多少钱。sell[i]表示卖i个最多有多少钱。

buy[j] = max(buy[j], sell[j-1] - prices[i]);  //看买prices[i]是否有原来划算
class Solution {
public:
    int maxProfit(int k, vector<int>& prices) {
        int n = prices.size(), i, j;
        if(n < 1)
            return 0;
        if(k >= (n>>1))
        {
            int ans = 0;
            for(i = 0; i < n-1; i++)
            {
                if(prices[i+1]-prices[i] > 0)
                    ans += prices[i+1]-prices[i];
            }
            return ans;
        }
        vector<vector<int>> dp(n, vector<int>(k+1, 0)); //dp[i][j]表示到第i天卖j个最多赚多少钱
        for(i = 1; i <= k; i++)
        {
            int buy = -prices[0];
            for(j = 0; j < n; j++)
            {
                dp[j][i] = max(j > 0 ? dp[j-1][i] : 0, buy + prices[j]);
                buy = max(buy, dp[j][i-1] - prices[j]);
            }
        }
        return dp[n-1][k];
    }
};

和上面一个算法思路一样。

121. 122. 123. 188. Best Time to Buy and Sell Stock *HARD* -- 买卖股票

原文:http://www.cnblogs.com/argenbarbie/p/5459311.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!