写在前面
本专栏专注于分析与讲解【面试经典150】算法,两到三天更新一篇文章,欢迎催更……
专栏内容以分析题目为主,并附带一些对于本题涉及到的数据结构等内容进行回顾与总结,文章结构大致如下,部分内容会有增删:
- Tag:介绍本题牵涉到的知识点、数据结构;
- 题目来源:贴上题目的链接,方便大家查找题目并完成练习;
- 题目解读:复述题目(确保自己真的理解题目意思),并强调一些题目重点信息;
- 解题思路:介绍一些解题思路,每种解题思路包括思路讲解、实现代码以及复杂度分析;
- 知识回忆:针对今天介绍的题目中的重点内容、数据结构进行回顾总结。
Tag
【一次遍历】【动态规划】【数组】【2023-10-01】
题目来源
![](https://i-blog.csdnimg.cn/blog_migrate/5a7c88f2fcea2b231ea850f85d56819c.png)
题目解读
给定一个表示股票价格的数组 prices
,prices[i]
表示股票第 i
天的价格,你可以选择在任意一天买入股票,并在以后的任意一天卖出该股票,求你能获得的最大利润。
解题思路
比较容易想到的方法是:枚举所有的买入和卖出价格,找出其中的最大差值即可。这样的时间复杂度为 O ( n 2 ) O(n^2) O(n2),对于数据量规模达到 1 0 5 10^5 105 的数据来说一定超时。
方法一:一次遍历
枚举所有的买入和卖出价格的方法超时,考虑一次遍历。
本题的重点是只买卖一次股票,那么我们肯定想在历史最低点买进股票,在历史最高点卖出股票(必须是买入时间的未来时间)。
于是想到维护一个变量 minVal
,表示已经遍历过价格的最小值(股票的历史最低点),遍历数组 prices
,那么我们在第 i
天(以第 i
天作为未来的历史最高点 )卖出股票的利润为 prices[i] - minVal
,我们返回遍历完所有天的最大利润即可,每次遍历时都需要更新当前的价格最小值。
复杂度分析
时间复杂度:
O
(
n
)
O(n)
O(n),
n
n
n 为数组 prices
的长度。
空间复杂度:
O
(
1
)
O(1)
O(1),除了答案 maxVal
之外,仅使用了一个额外的变量。
方法二:动态规划+空间优化
因为我们的手中最多有一张股票,我们每天可以选择买入或者卖出,所以每天交易结束后,我们手中要么还有一支股票,要么没有股票。
状态
dp[i][0]
表示第 i
天交易介绍后手中没有股票可以获得的最大利润,dp[i][1]
表示第 i
天交易介绍后手中有一支股票可以获得的最大利润。
转移关系
考虑 dp[i][0]
的转移方程,第 i
天交易结束后手中没有股票,可能前一天即 i-1
天已经没有股票了,对应的状态为 dp[i-1][0]
;也有可能是第 i-1
天手里有一支股票,第 i
天卖掉了,对应的状态为 pices[i] + dp[i-1][1]
。于是有转移关系:
d p [ i ] [ 0 ] = m a x ( d p [ i − 1 ] [ 0 ] , p r i c e s [ i ] + d p [ i − 1 ] [ 1 ] ) dp[i][0] = max(dp[i-1][0], prices[i] + dp[i-1][1]) dp[i][0]=max(dp[i−1][0],prices[i]+dp[i−1][1])
考虑 dp[i][1]
的转移方程,第 i
天交易介绍后手中有一支股票,可能前一天即 i-1
天手中就有一支股票,对应的状态为 dp[i-1][1]
;也有可能是第 i-1
天手里没有股票,第 i
天买入了一支股票,对应的状态为 -prices[i]
(因为只允许交易一次)。于是有转移关系:
d p [ i ] [ 1 ] = m a x ( d p [ i − 1 ] [ 1 ] , − p r i c e s [ i ] ) dp[i][1] = max(dp[i-1][1], -prices[i]) dp[i][1]=max(dp[i−1][1],−prices[i])
base case
初始状态,即第 0
天交易结束后有 dp[0][0] = 0
,dp[0][1] = -prices[0]
。
返回答案
返回答案为 dp[n-1][0]
,最后返回 max(dp[n-1][0], dp[n-1][1])
,因为最后手中没有股票一定比手中有股票收益高,所以直接返回 dp[n-1][0]
。
实现代码
class Solution {
public:
int maxProfit(vector<int>& prices) {
int n = prices.size();
vector<vector<int>> dp(n, vector<int>(2));
dp[0][1] = -prices[0];
for (int i = 1; i < n; ++i) {
dp[i][0] = max(dp[i-1][0], prices[i] + dp[i-1][1]);
dp[i][1] = max(dp[i-1][1], - prices[i]);
}
return dp[n-1][0];
}
};
空间优化
因为每天交易结束后不管手中还有没有股票,最大收益都和前一天结束后手中有没有股票的状态有关,因此可以使用两个变量来维护买股票(一天交易结束后手中还有股票)和卖股票(一天交易结束后手中没有股票)后手中剩余的最大钱数,也就是交易后的最大收益:
buy
:一天交易结束后手中还有股票的最大收益,初始化为-prices[0]
;sell
:一天交易结束后手中没有股票的最大收益,,初始化为0
。
最后返回的是经过 n
天交易后手中剩余的最大钱数,即返回 sell
。
实现代码
class Solution {
public:
int maxProfit(vector<int>& prices) {
int buy = -prices[0];
int sell = 0;
for (int p : prices) {
// 仅允许一次完整的交易(一次买+一次卖),因此上次卖出股票后手中的最大钱数为 0
buy = max(buy, 0 - p);
// 使用上次买入股票后手中的最大钱数 更新卖出股票后手中的钱
sell = max(sell, buy + p);
}
return sell;
}
};
复杂度分析
时间复杂度:
O
(
n
)
O(n)
O(n),
n
n
n 为数组 prices
的长度。
空间复杂度: O ( n ) O(n) O(n),优化后的空间复杂度为 O ( 1 ) O(1) O(1)。
写在最后
买卖股票系列题目
题目 | 解答 |
---|---|
122. 买卖股票的最佳时机 II | 【面试经典150】买卖股票的最佳时机 II |
123. 买卖股票的最佳时机 III | 【每日一题】买卖股票的最佳时机 III |
188. 买卖股票的最佳时机 IV | 【每日一题】买卖股票的最佳时机 IV |
309. 买卖股票的最佳时机含冷冻期 | 【每日一题】买卖股票的最佳时机含冷冻期 |
714. 买卖股票的最佳时机含手续费 | 【每日一题】买卖股票的最佳时机含手续费 |
如果文章内容有任何错误或者您对文章有任何疑问,欢迎私信博主或者在评论区指出 💬💬💬。
如果大家有更优的时间、空间复杂度方法,欢迎评论区交流。
最后,感谢您的阅读,如果感到有所收获的话可以给博主点一个 👍 哦。