Leetcode121-买卖股票的最佳时机-java、python
第一次写文章,记录自己的刷题。
题目链接:Leecode21
- 题目
给定一个数组,它的第 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(n2),时间复杂度较高。
采用一次遍历解决该问题,核心在于遍历过程中不断更新最低买入价格、最大利润,如下图所示。
引入一个变量inprice代表买入价格,对数组进行遍历,当数组元素低于inprice,则将inprice更改为数组元素。
引入一个变量res代表最大利润,计算以该数组元素作为卖出价格时的利润,当该利润大于res,则将res更改为最大利润。
遍历完成后,输出res。
(1) 采用一个变量代表买入点inprice
,采用res
代表最大利润。
(2) 遍历数组prices
,若prices[i]<inprice
,则将prices[i]
设为inprice
。将prices[i]
设为卖出点,则res = max(res,prices[i]−inprice)
。
一次循环遍历,时间复杂度O(n)。
引入两个变量inprice、res,空间复杂度O(1)。
- 代码
java
class Solution {
public int maxProfit(int[] prices) {
int inprice = Integer.MAX_VALUE;//买入点
int res=0;//利润
for(int i = 0;i<prices.length;i++){
if(prices[i]<inprice)
inprice = prices[i]; //更新买入点
res = Math.max(res,prices[i]-inprice);//当前最大利润
}
return res;
}
}
python
class Solution:
def maxProfit(self, prices: List[int]) -> int:
inprice = inf#买入点
res = 0#利润
for i in range(len(prices)):
if inprice>prices[i]:
inprice = prices[i]#更新买入点
res = max(res,prices[i]-inprice);#当前最大利润
return res