Best Time to Buy and Sell Stock

Question

  1. Say you have an array for
  2. which the ith element is the price of a given stock on day i.
  4. If you were only permitted to complete at most one transaction
  5. (ie, buy one and sell one share of the stock),
  6. design an algorithm to find the maximum profit.
  8. Example
  9. Given an example [3,2,3,1,2], return 1

题解

最多只允许进行一次交易，显然我们只需要把波谷和波峰分别找出来就好了。但是这样的话问题又来了，有多个波峰和波谷时怎么办？——找出差值最大的一对波谷和波峰。故需要引入一个索引用于记录当前的波谷，结果即为当前索引值减去波谷的值。

Python

  1. class Solution:
  2.     """
  3.     @param prices: Given an integer array
  4.     @return: Maximum profit
  5.     """
  6.     def maxProfit(self, prices):
  7.         if prices is None or len(prices) <= 1:
  8.             return 0
  10.         profit = 0
  11.         cur_price_min = 2**31 - 1
  12.         for price in prices:
  13.             profit = max(profit, price - cur_price_min)
  14.             cur_price_min = min(cur_price_min, price)
  16.         return profit

C++

  1. class Solution {
  2. public:
  3.     /**
  4.      * @param prices: Given an integer array
  5.      * @return: Maximum profit
  6.      */
  7.     int maxProfit(vector<int> &prices) {
  8.         if (prices.size() <= 1) return 0;
  10.         int profit = 0;
  11.         int cur_price_min = INT_MAX;
  12.         for (int i = 0; i < prices.size(); ++i) {
  13.             profit = max(profit, prices[i] - cur_price_min);
  14.             cur_price_min = min(cur_price_min, prices[i]);
  15.         }
  17.         return profit;
  18.     }
  19. };

Java

  1. public class Solution {
  2.     /**
  3.      * @param prices: Given an integer array
  4.      * @return: Maximum profit
  5.      */
  6.     public int maxProfit(int[] prices) {
  7.         if (prices == null || prices.length <= 1) return 0;
  9.         int profit = 0;
  10.         int curPriceMin = Integer.MAX_VALUE;
  11.         for (int price : prices) {
  12.             profit = Math.max(profit, price - curPriceMin);
  13.             curPriceMin = Math.min(curPriceMin, price);
  14.         }
  16.         return profit;
  17.     }
  18. }

源码分析

善用max和min函数，减少if的使用。

复杂度分析

遍历一次 prices 数组，时间复杂度为 O(n), 使用了几个额外变量，空间复杂度为 O(1).

Reference

  • soulmachine 的卖股票系列