1. 程式人生 > 其它 >leetcode——買賣股票的最佳時機

leetcode——買賣股票的最佳時機

技術標籤:面試C++

題目:

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

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

注意:你不能在買入股票前賣出股票。
連結:https://leetcode-cn.com/problems/best-time-to-buy-and-sell-stock

示例:

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。

方案一:暴力求解

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

時間複雜度O(n^2)

空間複雜度O(1)

方案二:一次遍歷

在歷史最低點買入,當天的收益會最大。然後取每天可以取得的最大收益的最大值。

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        int profit = 0;
        int lowPrice = INT_MAX;
        int len = prices.size();
        for (int i = 0; i < len; i++) {
            if (i > 0) {
                int p = prices[i] - lowPrice;
                if (p > profit) profit = p;
            }
            if (prices[i] < lowPrice) {
                lowPrice = prices[i];
            }
        }
        return profit;
    }
};

時間複雜度O(n)

空間複雜度O(1)