力扣题目
解题思路
java代码
力扣题目:
给你一个整数数组 prices
,其中 prices[i]
表示某支股票第 i
天的价格。
在每一天,你可以决定是否购买和/或出售股票。你在任何时候 最多 只能持有 一股 股票。你也可以先购买,然后在 同一天 出售。
返回 你能获得的 最大 利润 。
示例 1:
输入:prices = [7,1,5,3,6,4] 输出:7 解释:在第 2 天(股票价格 = 1)的时候买入,在第 3 天(股票价格 = 5)的时候卖出, 这笔交易所能获得利润 = 5 - 1 = 4。 随后,在第 4 天(股票价格 = 3)的时候买入,在第 5 天(股票价格 = 6)的时候卖出, 这笔交易所能获得利润 = 6 - 3 = 3。 最大总利润为 4 + 3 = 7 。
示例 2:
输入:prices = [1,2,3,4,5] 输出:4 解释:在第 1 天(股票价格 = 1)的时候买入,在第 5 天 (股票价格 = 5)的时候卖出, 这笔交易所能获得利润 = 5 - 1 = 4。 最大总利润为 4 。
示例 3:
输入:prices = [7,6,4,3,1] 输出:0 解释:在这种情况下, 交易无法获得正利润,所以不参与交易可以获得最大利润,最大利润为 0。
解题思路:
算法原理:
- 使用动态规划的方法。定义两个状态:持有股票(
dp[i][1]
)和不持有股票(dp[i][0]
)。通过递推公式来更新每个状态下的最大利润。
思路:
- 在第
i
天,不持有股票的最大利润有两种情况:要么是前一天就不持有且保持,即dp[i-1][0]
;要么是前一天持有股票,在今天卖出,即dp[i-1][1]+prices[i]
,取两者中的最大值。 - 持有股票的最大利润也有两种情况:要么是前一天就持有且保持,即
dp[i-1][1]
;要么是前一天不持有,在今天买入,即dp[i-1][0]-Prices[i]
,取两者中的最大值。
代码分析:
- 初始化
dp[0][0]
为 0,表示第一天不持有股票的利润为 0;dp[0][1]
为-prices[0]
,表示第一天持有股票的成本为买入价格。 - 通过循环从第二天开始计算每个状态下的利润,根据递推公式更新
dp[i][0]
和dp[i][1]
。 - 最后返回最后一天不持有股票的最大利润
dp[n-1][0]
。
时间复杂度:O(n)。
空间复杂度:O(n),使用了二维数组来存储状态。可以通过优化空间复杂度到 O(1)。
java代码:
package org.example.mouth8;
public class Leetcode122 {
public static void main(String[] args) {
int[] prices = {7, 1, 5, 3, 6, 4};
int[] prices2 = {1, 2, 3, 4, 5};
System.out.println(maxProfit(prices2));
System.out.println(maxProfit(prices));
}
public static int maxProfit(int[] prices) {
int n = prices.length;
int[][] dp = new int[n][2];
dp[0][0] = 0;
dp[0][1] = -prices[0];
for (int i = 1; i < n; i++) {
dp[i][0] = Math.max(dp[i - 1][0], dp[i - 1][1] + prices[i]);
dp[i][1] = Math.max(dp[i - 1][1], dp[i - 1][0] - prices[i]);
}
return dp[n - 1][0];
}
}
更多详细内容同步到公众号,感谢大家的支持!
每天都会给刷算法的小伙伴推送明日一题,并且没有任何收费项