官方题解二刚开始没看太明白,看了评论区的思路才想出来。
题目描述
给定一个数组 prices ,它的第 i 个元素 prices[i] 表示一支给定股票第 i 天的价格。
你只能选择某一天买入这只股票,并选择在未来的某一个不同的日子卖出该股票。设计一个算法来计算你所能获取的最大利润。
返回你可以从这笔交易中获取的最大利润。如果你不能获取任何利润,返回0
。
示例 1:
输入:
[7,1,5,3,6,4]
输出:5
解释:在第 2 天(股票价格 = 1)的时候买入,在第 5 天(股票价格 = 6)的时候卖出,最大利润 = 6-1 = 5 。
注意利润不能是 7-1 = 6, 因为卖出价格需要大于买入价格;同时,你不能在买入前卖出股票。
示例 2:
输入:
prices = [7,6,4,3,1]
输出:0
解释:在这种情况下, 没有交易完成, 所以最大利润为 0。
提示:
1 <= prices.length <= 105
0 <= prices[i] <= 104
思路
暴力解法超时,题目提示动态规划,那就考虑状态转移方程。
maxProfit[i] = prices[i] - minPrices[i]
其中
maxProfit[i]
为第i
天卖出股票时能得到的最大利润
由上式可知,第i
天卖出股票能得到的最大利润一定是第i
天的价格与第i
天之前的最低价之差。
维护一个最低价数组minPrices[]
,i
天的最低价与i-1
天的最低价有关,得到转移方程
minPrices[i] = min(minPrices[i-1], prices[i])
maxProfit = max(maxProfit, prices[i] - minPrices[i]
* 优化
原思路的时间复杂度为O(n)
,空间复杂度为O(n)
观察后知,minPrices[i]
实际上只与minPrices[i-1]
有关,那么可以仅用一个变量去记录minPrice
minPrice = min(minPrice, prices[i])
题解
class Solution {
public:
int maxProfit(vector<int>& prices)
{
int ret = 0;
int minPrice = prices[0];
for(int i=1; i<prices.size(); i++)
{
minPrice = min(minPrice, prices[i]);
int temp = prices[i] - minPrice;
ret = temp > ret ? temp : ret;
}
return ret;
}
};