Best Time to Buy and Sell Stock - LeetCode

https://leetcode.com/problems/best-time-to-buy-and-sell-stock/?envType=study-plan-v2&envId=top-interview-150

LeetCode_Sharing.png

You are given an array prices where prices[i] is the price of a given stock on the ith day.

You want to maximize your profit by choosing a single day to buy one stock and choosing a different day in the future to sell that stock.

Return the maximum profit you can achieve from this transaction. If you cannot achieve any profit, return 0.

Example 1:

Input: prices = [7,1,5,3,6,4]
Output: 5
Explanation: Buy on day 2 (price = 1) and sell on day 5 (price = 6), profit = 6-1 = 5.
Note that buying on day 2 and selling on day 1 is not allowed because you must buy before you sell.

Example 2:

Input: prices = [7,6,4,3,1]
Output: 0
Explanation: In this case, no transactions are done and the max profit = 0.

Constraints:

  • 1 <= prices.length <= 105
  • 0 <= prices[i] <= 104

解析

直观的办法是把每天作为买入起点,在每一次内计算最大值,最终全盘比较。

class Solution:
    def maxProfit(self, prices: List[int]) -> int:
        profit = 0
        for i in range(len(prices) - 1):
            max_profit = 0
            for j in range(i + 1, len(prices)):
                if max_profit < prices[j] - prices[i]:
                    max_profit = prices[j] - prices[i]
            if max_profit > profit:
                profit = max_profit

        return profit

结果是超时了,两个循环嵌套的必然结果。不做讨论了,想想其他办法。

下面的方法更好一些,从前往后循环,一个变量存储当前的最小值,并在此基础上计算Max profit。

class Solution:
    def maxProfit(self, prices: List[int]) -> int:
        min_price = prices[0]
        max_profit = 0
        for i in range(len(prices)):
            if prices[i] < min_price:
                min_price = prices[i]
            elif prices[i] - min_price > max_profit:
                max_profit = prices[i] - min_price

        return max_profit
class Solution {
    public int maxProfit(int[] prices) {
        int max_profit = 0;
        int min_price = prices[0];

        for (int price:prices) {
            if (price < min_price){
                min_price = price;
            } else if (max_profit < price - min_price) {
                max_profit = price - min_price;
            }
        }
        return max_profit;
    }
}
class Solution {
public:
    int maxProfit(vector<int>& prices) {
        int max_profit = 0;
        int min_price = prices[0];

        for (int price : prices){
            if (price < min_price){
                min_price = price;
            }
            else if (max_profit < price - min_price){
                max_profit = price - min_price;
            }
        }
        return max_profit;
    }
};