[LeetCode] 買賣股票的最佳時機1-3

買賣股票的最佳時機


121、買賣股票的最佳時機

難度:Easy

題目描述:

給定一個數組,它的第 i 個元素是一支給定股票第 i 天的價格。

如果你最多隻允許完成一筆交易(即買入和賣出一支股票一次),設計一個算法來計算你所能獲取的最大利潤。

注意:你不能在買入股票前賣出股票。

示例 1:

輸入: [7,1,5,3,6,4]
輸出: 5
解釋: 在第 2 天(股票價格 = 1)的時候買入,在第 5 天(股票價格 = 6)的時候賣出,最大利潤 = 6-1 = 5 。
注意利潤不能是 7-1 = 6, 因爲賣出價格需要大於買入價格;同時,你不能在買入前賣出股票。

示例 2:

輸入: [7,6,4,3,1]
輸出: 0
解釋: 在這種情況下, 沒有交易完成, 所以最大利潤爲 0。

解題思路:

解法一:暴力法

因爲只能進行一次買賣,所以可以暴力遍歷所有的情況。
對於每一天,遍歷其後面的每一天,記錄兩天之差的最大值。
時間複雜度O(N^2),空間複雜度O(1)。注意這種複雜度下會超時。

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

        return max;
    }
};

解法二:動態規劃

dp[i]表示從第1天到第i天的最大利潤,min表示當前的股票最低值。
狀態轉移方程:dp[i] = max(dp[i - 1], price[i] - min)
進一步優化,考慮到dp[i]只與dp[i-1]有關,所以可以僅用一個變量maxProfit表示。
空間複雜度O(N),時間複雜度O(1)。

class Solution {
public:
    int maxProfit(vector<int>& prices) {

        if (prices.size() <= 1) return 0;

        int minPrice = __INT_MAX__;
        int maxProfit = 0;

        for (int i = 0; i < prices.size(); i++)
        {
            minPrice = min(minPrice, prices[i]);
            maxProfit = max(maxProfit, prices[i] - minPrice);
        }
        
        return maxProfit;
    }
};

122、買賣股票的最佳時機-2

難度:Easy

題目描述:

給定一個數組,它的第 i 個元素是一支給定股票第 i 天的價格。

設計一個算法來計算你所能獲取的最大利潤。你可以儘可能地完成更多的交易(多次買賣一支股票)。

注意:你不能同時參與多筆交易(你必須在再次購買前出售掉之前的股票)。

示例 1:

輸入: [7,1,5,3,6,4]
輸出: 7
解釋: 在第 2 天(股票價格 = 1)的時候買入,在第 3 天(股票價格 =5)的時候賣出, 這筆交 易所能獲得利潤 = 5-1 = 4 。
隨後,在第 4 天(股票價格 = 3)的時候買入,在第 5 天(股票價格 = 6)的時候賣出, 這筆交易所能獲得利潤 = 6-3 = 3 。

示例 2:

輸入: [1,2,3,4,5]
輸出: 4 解釋: 在第 1 天(股票價格 = 1)的時候買入,在第 5 天 (股票價格 = 5)的時候賣出, 這筆交易所能獲得利潤 = 5-1 = 4 。
注意你不能在第 1 天和第 2 天接連購買股票,之後再將它們賣出。
因爲這樣屬於同時參與了多筆交易,你必須在再次購買前出售掉之前的股票。

示例 3:

輸入: [7,6,4,3,1]
輸出: 0
解釋: 在這種情況下, 沒有交易完成, 所以最大利潤爲 0。

提示:

1 <= prices.length <= 3 * 10 ^ 4
0 <= prices[i] <= 10 ^ 4

解題思路:

對比上一題,這道題的特點是可以無數次交易。
因此這道題的思路就是貪心法,每當後一天的價錢高於前一天,就執行一次交易。

class Solution {
public:
    int maxProfit(vector<int>& prices) {

        if (prices.size() <= 1) return 0;

        int result = 0;

        for (size_t i = 1; i < prices.size(); i++)
        {
            if (prices[i-1] < prices[i])
            {
                result += prices[i] - prices[i-1];
            }
        }
        
        return result;
    }
};

123、買賣股票的最佳時機-3

難度:Hard

題目描述:

給定一個數組,它的第 i 個元素是一支給定的股票在第 i 天的價格。

設計一個算法來計算你所能獲取的最大利潤。你最多可以完成 兩筆 交易。

注意: 你不能同時參與多筆交易(你必須在再次購買前出售掉之前的股票)。

示例 1:

輸入: [3,3,5,0,0,3,1,4]
輸出: 6
解釋: 在第 4 天(股票價格 = 0)的時候買入,在第 6 天(股票價格 = 3)的時候賣出,這筆交易所能獲得利潤 = 3-0 = 3 。
隨後,在第 7 天(股票價格 = 1)的時候買入,在第 8 天 (股票價格 = 4)的時候賣出,這筆交易所能獲得利潤 = 4-1 = 3 。

示例 2:

輸入: [1,2,3,4,5]
輸出: 4 解釋: 在第 1 天(股票價格 = 1)的時候買入,在第 5 天 (股票價格 = 5)的時候賣出, 這筆交易所能獲得利潤 = 5-1 = 4 。
注意你不能在第 1 天和第 2 天接連購買股票,之後再將它們賣出。
因爲這樣屬於同時參與了多筆交易,你必須在再次購買前出售掉之前的股票。

示例 3:

輸入: [7,6,4,3,1]
輸出: 0
解釋: 在這個情況下, 沒有交易完成, 所以最大利潤爲 0。

解題思路:

與上面的題相比,難點在於規定了可以交易兩次,並且必須先買後賣。
因此現有有三個狀態共同決定當前最大利潤:當前的天數i(0-n), 已經買入的次數j(0-2),當前是否持有股票(0-1)。

所以利用動態規劃,狀態轉移方程如下:

dp[i][j][0]=max(dp[i1][j][0],dp[i1][j][1]+prices[i1])dp[i][j][0] = max(dp[i-1][j][0], dp[i-1][j][1] + prices[i-1])

dp[i][j][1]=max(dp[i1][j][1],dp[i1][j1][0]prices[i1])dp[i][j][1] = max(dp[i-1][j][1], dp[i-1][j-1][0] - prices[i-1])

注意,由於天數是(0-n),所以第i天時,對應的股票價格是prices[i-1]

最終的輸出爲:
max(dp[n][2][0], dp[n][1][0], dp[n][0][0])

初始值:
dp[i][0][0] = 0
dp[i][0][1] = -inf
dp[0][i][0] = -inf
dp[0][i][1] = -inf

class Solution {
public:
    int maxProfit(vector<int>& prices) {

        const int inf = 1 << 30;
        const int n = prices.size();
        int dp[30000 + 5][3][2];

        for (int i = 0; i <= n; i++)
        {
            dp[i][0][0] = 0;
            dp[i][0][1] = -inf;
        }
        for (int i = 1; i < 3; i++)
        {
            dp[0][i][0] = -inf;
            dp[0][i][1] = -inf;
        }

        for (int i = 1; i <= n; i++)
        {
            for (int j = 0; j < 3; j++)
            {
                dp[i][j][0] = max(dp[i-1][j][0], dp[i-1][j][1] + prices[i-1]);
                if (j)
                {
                    dp[i][j][1] = max(dp[i-1][j][1], dp[i-1][j-1][0] - prices[i-1]);
                }  
            }
        }
        
        
        return max(max(dp[n][0][0], dp[n][1][0]), dp[n][2][0]);
        
    }
};
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章