坏蛋Dan
知乎@坏蛋Dan
发布时间:2024.2.7

前言

今天也有两道题,一道简单,另一道是变体,中等难度


题目1

给定一个数组 prices ,它的第 i 个元素 prices[i] 表示一支给定股票第 i 天的价格。

你只能选择 某一天 买入这只股票,并选择在 未来的某一个不同的日子 卖出该股票。设计一个算法来计算你所能获取的最大利润。

返回你可以从这笔交易中获取的最大利润。如果你不能获取任何利润,返回 0

示例 1:

示例 2:

提示:

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

分析:

这道题说难也不难,说简单的话也不简单,因为暴力方案也就是时间复杂度为O(n ^ 2)的方案是超时的 所以我们得思考一下小于O(n ^ 2)时间复杂度的方案。 前面我们说过:当给的数据是一维的,那基本上就意味着存在O(n)的方案,因为是算法题,要的就是巧妙地方案,所以官方出的题都是有骚解法的。

你应该注意到了最大利润里的最大, 那我们是否可以使用动态规划来实现呢? 理论上都是可以的 那么我们要找的目标是啥?求什么设什么,那么我们设dp[i]最大利润,那么久就存在dp[i] = max(dp[i, dp[i - 1]])。 接着我们要找关系 而利润是售出金额 - 收入金额。 所以dp[i]prices[i] - cost,有关。 而我们要的是最大的利润,所以cost必须是最小的,即min(cost, prices[i]) 所以dp[i] = max(dp[i - 1], prices[i] - min(cost, prices[i]))

然后我们还可以优化下,没必要存储一个数组记录每个元素的最大利润,我们只比较前后两个,所以一个max变量即可。

那么我们就需要两个变量:cost,用来记录一个最大利润:earn

然后处理初始值,由于我们需要dp[i - 1],所以我们需要知道第一个元素的值,所以起始下标是1


代码就没啥好说的了


题目2

给你一个整数数组 prices ,其中 prices[i] 表示某支股票第 i 天的价格。

在每一天,你可以决定是否购买和/或出售股票。你在任何时候 最多 只能持有 一股 股票。你也可以先购买,然后在 同一天 出售。

返回 你能获得的 最大 利润

示例 1:

示例 2:

示例 3:

提示:

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

分析

这道题和上面那道也可以用动态规划来解,但是我们还有更快的方案。 我们不再是只能买一次,我们可以买多次,不过同一时刻只能有一个。 所求的是总和最多。

那我们可以啊,只要第一天买入第二天卖出的差值大于0,那就是不亏,不亏就是赚啊!

然后我们再来看个老哥的写法

简单地说就是使用windows切割成长度为2,间隔为1的二维数组(迭代器), 比如7, 1, 5, 3, 6, 4变成[[7, 1], 1, 5, 5, 3, 3, 6, 6, 4] 然后去计算每个切片的差值变成一个新数组:0, 4, 0, 3, 0,最后相加,正好是7,

写法是很优雅,但是耗时和空间复杂度较高


测试用例


总结

第一道简单的题实际上不算简单,因为暴力方案不可用,所以得绕下弯子。 如果没学过动态规划,那就比较难了。 第二道题则是要找到重点:可多次买入卖出