Java+Leetcode+动态规划VII
·
122.买卖股票的最佳时机II
给你一个整数数组
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。提示:
1 <= prices.length <= 3 * 1040 <= prices[i] <= 104
原理
- 定义状态:
- 使用二维数组 dp,其中:
- dp[i][0]:第 i 天结束时不持有股票的最大利润。
- dp[i][1]:第 i 天结束时持有股票的最大利润。
- 目标是求 dp[n-1][0],因为最后一天不持有股票(卖出或不交易)会产生最大利润。
- 初始化:
- 第0天:
- dp[0][0] = 0:不持有股票,利润为0。
- dp[0][1] = -prices[0]:持有股票,利润为负的买入价格。
- 状态转移:
- 对于每一天 i(从1到 n-1):
- 持有股票(dp[i][1]):
- 继续持有:dp[i-1][1]。
- 前一天不持有,当天买入:dp[i-1][0] - prices[i]。
- 取最大值:dp[i][1] = max(dp[i-1][1], dp[i-1][0] - prices[i])。
- 不持有股票(dp[i][0]):
- 继续不持有:dp[i-1][0]。
- 前一天持有,当天卖出:dp[i-1][1] + prices[i]。
- 取最大值:dp[i][0] = max(dp[i-1][0], dp[i-1][1] + prices[i])。
- 允许同一天买卖,因为状态转移基于前一天状态,买卖操作在同一天内可合并。
- 优化可能性:
- 状态只依赖前一天,可以用两个变量(hold 和 notHold)代替 dp 数组,优化空间到 O(1)。
- 贪心算法更简单:累加所有正价格差(prices[i] - prices[i-1]),但这里使用动态规划。
- 时间复杂度:
- 遍历数组:O(n),n 为 prices.length。
- 每次操作(比较、加减):O(1)。
- 总时间复杂度:O(n)。
- 空间复杂度:
- 使用二维数组 dp[n][2]:O(n)。
代码
class Solution {
public int maxProfit(int[] prices) {
int n = prices.length;
// 创建二维dp数组,dp[i][0]表示第i天不持有股票的最大利润,dp[i][1]表示持有股票的最大利润
int[][] dp = new int[n][2];
// 初始化第0天状态
dp[0][0] = 0; // 第0天不持有股票,利润为0
dp[0][1] = -prices[0]; // 第0天持有股票,利润为负的买入价格
// 遍历从第1天到最后一天
for (int i = 1; i < n; ++i) {
// 持有股票的最大利润:
// 1. 前一天已持有(dp[i-1][1])
// 2. 前一天不持有,当天买入(dp[i-1][0] - prices[i])
dp[i][1] = Math.max(dp[i - 1][1], dp[i - 1][0] - prices[i]);
// 不持有股票的最大利润:
// 1. 前一天不持有(dp[i-1][0])
// 2. 前一天持有,当天卖出(dp[i-1][1] + prices[i])
dp[i][0] = Math.max(dp[i - 1][0], dp[i - 1][1] + prices[i]);
}
// 返回最后一天不持有股票的最大利润
return dp[n - 1][0];
}
}
123.买卖股票的最佳时机III
给定一个数组,它的第
i个元素是一支给定的股票在第i天的价格。设计一个算法来计算你所能获取的最大利润。你最多可以完成 两笔 交易。
注意:你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。
示例 1:
输入:prices = [3,3,5,0,0,3,1,4] 输出:6 解释:在第 4 天(股票价格 = 0)的时候买入,在第 6 天(股票价格 = 3)的时候卖出,这笔交易所能获得利润 = 3-0 = 3 。 随后,在第 7 天(股票价格 = 1)的时候买入,在第 8 天 (股票价格 = 4)的时候卖出,这笔交易所能获得利润 = 4-1 = 3 。示例 2:
输入:prices = [1,2,3,4,5] 输出:4 解释:在第 1 天(股票价格 = 1)的时候买入,在第 5 天 (股票价格 = 5)的时候卖出, 这笔交易所能获得利润 = 5-1 = 4 。 注意你不能在第 1 天和第 2 天接连购买股票,之后再将它们卖出。 因为这样属于同时参与了多笔交易,你必须在再次购买前出售掉之前的股票。示例 3:
输入:prices = [7,6,4,3,1] 输出:0 解释:在这个情况下, 没有交易完成, 所以最大利润为 0。示例 4:
输入:prices = [1] 输出:0提示:
1 <= prices.length <= 1050 <= prices[i] <= 105
原理
- 定义状态:
- 使用二维数组 dp[i][j],表示第 i 天处于状态 j 的最大利润:
- j = 0:无操作(默认0,未使用)。
- j = 1:第一次持有股票。
- j = 2:第一次卖出股票。
- j = 3:第二次持有股票。
- j = 4:第二次卖出股票。
- 目标是 dp[n-1][4],表示最后一天完成最多两次交易的最大利润。
- 初始化:
- 第0天:
- dp[0][1] = -prices[0]:第一次买入,利润为负价格。
- dp[0][3] = -prices[0]:第二次买入(理论上不可达,但初始化为负价格)。
- dp[0][0] = 0, dp[0][2] = 0, dp[0][4] = 0(默认)。
- 如果 n = 1,无法交易,返回0。
- 状态转移:
- 对于每一天 i(从1到 n-1):
- 第一次持有(dp[i][1]):
- 继续持有:dp[i-1][1]。
- 当天买入:-prices[i](假设从无操作状态开始)。
- 取最大值:dp[i][1] = max(dp[i-1][1], -prices[i])。
- 第一次卖出(dp[i][2]):
- 继续不持有:dp[i-1][2]。
- 当天卖出:dp[i-1][1] + prices[i]。
- 取最大值:dp[i][2] = max(dp[i-1][2], dp[i-1][1] + prices[i])。
- 第二次持有(dp[i][3]):
- 继续持有:dp[i-1][3]。
- 第一次卖出后当天买入:dp[i-1][2] - prices[i]。
- 取最大值:dp[i][3] = max(dp[i-1][3], dp[i-1][2] - prices[i])。
- 第二次卖出(dp[i][4]):
- 继续不持有:dp[i-1][4]。
- 当天卖出:dp[i-1][3] + prices[i]。
- 取最大值:dp[i][4] = max(dp[i-1][4], dp[i-1][3] + prices[i])。
- 状态 j=0 未使用,始终为0。
- 优化可能性:
- 可以用一维数组(5个变量)代替 dp 数组,优化空间到 O(1),因为状态只依赖前一天。
- 当前代码使用二维数组,逻辑更直观。
- 时间复杂度:
- 遍历数组:O(n),n 为 prices.length。
- 每次更新5个状态:O(1)。
- 总时间复杂度:O(n)。
- 空间复杂度:
- 使用二维数组 dp[n][5]:O(n)。
代码
import java.util.*;
class Solution {
public int maxProfit(int[] prices) {
// 特殊情况:如果只有一天,无法交易,返回0
if (prices.length == 1) return 0;
// 创建二维dp数组,dp[i][j]表示第i天处于状态j的最大利润
// 状态j:0(无操作),1(第一次持有),2(第一次卖出),3(第二次持有),4(第二次卖出)
int[][] dp = new int[prices.length][5];
// 初始化第0天状态
dp[0][1] = -prices[0]; // 第一次买入
dp[0][3] = -prices[0]; // 第二次买入(假设先买入,但实际不可达,仅初始化)
// dp[0][0], dp[0][2], dp[0][4] 默认0(无操作、卖出状态)
// 遍历从第1天到最后一天
for (int i = 1; i < prices.length; i++) {
// 第一次持有:继续持有或当天买入
dp[i][1] = Math.max(dp[i-1][1], -prices[i]);
// 第一次卖出:继续不持有或当天卖出
dp[i][2] = Math.max(dp[i-1][2], dp[i-1][1] + prices[i]);
// 第二次持有:继续持有或在第一次卖出后当天买入
dp[i][3] = Math.max(dp[i-1][3], dp[i-1][2] - prices[i]);
// 第二次卖出:继续不持有或当天卖出
dp[i][4] = Math.max(dp[i-1][4], dp[i-1][3] + prices[i]);
// 调试输出:打印当前天的状态
System.out.print(dp[i][1] + " " + dp[i][2] + " " + dp[i][3] + " " + dp[i][4] + "\n");
}
// 返回最后一天第二次卖出的最大利润
return dp[prices.length - 1][4];
}
}
188.买卖股票的最佳时机IV
给你一个整数数组
prices和一个整数k,其中prices[i]是某支给定的股票在第i天的价格。设计一个算法来计算你所能获取的最大利润。你最多可以完成
k笔交易。也就是说,你最多可以买k次,卖k次。注意:你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。
示例 1:
输入:k = 2, prices = [2,4,1] 输出:2 解释:在第 1 天 (股票价格 = 2) 的时候买入,在第 2 天 (股票价格 = 4) 的时候卖出,这笔交易所能获得利润 = 4-2 = 2 。示例 2:
输入:k = 2, prices = [3,2,6,5,0,3] 输出:7 解释:在第 2 天 (股票价格 = 2) 的时候买入,在第 3 天 (股票价格 = 6) 的时候卖出, 这笔交易所能获得利润 = 6-2 = 4 。 随后,在第 5 天 (股票价格 = 0) 的时候买入,在第 6 天 (股票价格 = 3) 的时候卖出, 这笔交易所能获得利润 = 3-0 = 3 。提示:
1 <= k <= 1001 <= prices.length <= 10000 <= prices[i] <= 1000
原理
- 定义状态:
- 使用二维数组 dp[i][j],表示第 i 天处于状态 j 的最大利润:
- j = 0:无操作。
- j = 1:第一次持有股票。
- j = 2:第一次卖出股票。
- j = 3:第二次持有股票。
- j = 4:第二次卖出股票。
- ...
- j = 2*k-1:第 k 次持有股票。
- j = 2*k:第 k 次卖出股票。
- 目标是 dp[n-1][2*k],表示最后一天完成最多 k 次交易的最大利润。
- 初始化:
- 第0天:
- 奇数状态(j = 1, 3, 5, ...):第1, 2, 3...次买入,dp[0][2*i+1] = -prices[0]。
- 偶数状态(j = 0, 2, 4, ...):无操作或第1, 2...次卖出,默认0。
- 如果 n = 1,无法交易,返回0。
- 状态转移:
- 对于每一天 i(从1到 n-1):
- 第 j 次持有(dp[i][2*j-1]):
- 继续持有:dp[i-1][2*j-1]。
- 前一天为第 j-1 次卖出(或无操作,j=1 时),当天买入:dp[i-1][2*j-2] - prices[i]。
- 取最大值:dp[i][2*j-1] = max(dp[i-1][2*j-1], dp[i-1][2*j-2] - prices[i])。
- 第 j 次卖出(dp[i][2*j]):
- 继续不持有:dp[i-1][2*j]。
- 前一天为第 j 次持有,当天卖出:dp[i-1][2*j-1] + prices[i]。
- 取最大值:dp[i][2*j] = max(dp[i-1][2*j], dp[i-1][2*j-1] + prices[i])。
- 优化可能性:
- 当 k 较大(例如 k >= n/2),问题等价于“买卖股票 II”(无限次交易),可用贪心算法。
- 空间可优化到 O(1),用一维数组 dp[2*k+1] 替代 dp[n][2*k+1],因为状态只依赖前一天。
- 时间复杂度:
- 遍历天数:O(n),n 为 prices.length。
- 遍历状态:O(k)。
- 总时间复杂度:O(n * k)。
- 空间复杂度:
- 使用二维数组 dp[n][2*k+1]:O(n * k)。
代码
class Solution {
public int maxProfit(int k, int[] prices) {
// 特殊情况:如果只有一天,无法交易,返回0
if (prices.length == 1) return 0;
// 创建二维dp数组,dp[i][j]表示第i天处于状态j的最大利润
// 状态j:0(无操作),1(第一次持有),2(第一次卖出),3(第二次持有),4(第二次卖出)... 2*k(第k次卖出)
int[][] dp = new int[prices.length][2 * k + 1];
// 初始化第0天状态
// 奇数状态(1,3,5...)表示第1,2,3...次持有股票,初始化为负的买入价格
for (int i = 0; i < k; i++) {
dp[0][2 * i + 1] = -prices[0];
}
// 偶数状态(0,2,4...)表示无操作或第1,2...次卖出,默认0
// 遍历从第1天到最后一天
for (int i = 1; i < prices.length; i++) {
// 遍历k次交易的每个状态
for (int j = 1; j <= k; j++) {
// 第j次持有股票(状态2*j-1):
// 1. 继续持有:dp[i-1][2*j-1]
// 2. 前一天为第j-1次卖出(或无操作,j=1时),当天买入:dp[i-1][2*j-2] - prices[i]
dp[i][2 * j - 1] = Math.max(dp[i - 1][2 * j - 1], dp[i - 1][2 * j - 2] - prices[i]);
// 第j次卖出股票(状态2*j):
// 1. 继续不持有:dp[i-1][2*j]
// 2. 前一天为第j次持有,当天卖出:dp[i-1][2*j-1] + prices[i]
dp[i][2 * j] = Math.max(dp[i - 1][2 * j], dp[i - 1][2 * j - 1] + prices[i]);
}
}
// 返回最后一天第k次卖出的最大利润
return dp[prices.length - 1][2 * k];
}
}
309.买卖股票的最佳时机含冷冻期
给定一个整数数组
prices,其中第prices[i]表示第i天的股票价格 。设计一个算法计算出最大利润。在满足以下约束条件下,你可以尽可能地完成更多的交易(多次买卖一支股票):
- 卖出股票后,你无法在第二天买入股票 (即冷冻期为 1 天)。
注意:你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。
示例 1:
输入: prices = [1,2,3,0,2] 输出: 3 解释: 对应的交易状态为: [买入, 卖出, 冷冻期, 买入, 卖出]示例 2:
输入: prices = [1] 输出: 0提示:
1 <= prices.length <= 50000 <= prices[i] <= 1000
原理
- 定义状态:
- 使用二维数组 dp[i][j],表示第 i 天处于状态 j 的最大利润:
- j = 0:持有股票。
- j = 1:不持有股票且不在冷冻期(可买入)。
- j = 2:当天卖出股票。
- j = 3:处于冷冻期(不可买入)。
- 目标是最后一天不持有股票的最大利润:max(dp[n-1][1], dp[n-1][2], dp[n-1][3])。
- 初始化:
- 第0天:
- dp[0][0] = -prices[0]:持有股票(买入)。
- dp[0][1] = 0:不持有且不在冷冻期。
- dp[0][2] = 0:卖出(不可达,默认0)。
- dp[0][3] = 0:冷冻期(不可达,默认0)。
- 状态转移:
- 对于每一天 i(从1到 n-1):
- 持有股票(dp[i][0]):
- 继续持有:dp[i-1][0]。
- 前一天可买入,当天买入:dp[i-1][1] - prices[i]。
- 前一天冷冻期结束,当天买入:dp[i-1][3] - prices[i]。
- 取最大值:dp[i][0] = max(dp[i-1][0], dp[i-1][1] - prices[i], dp[i-1][3] - prices[i])。
- 不持有且不在冷冻期(dp[i][1]):
- 继续不持有:dp[i-1][1]。
- 前一天冷冻期结束:dp[i-1][3]。
- 取最大值:dp[i][1] = max(dp[i-1][1], dp[i-1][3])。
- 当天卖出(dp[i][2]):
- 前一天持有,当天卖出:dp[i-1][0] + prices[i]。
- dp[i][2] = dp[i-1][0] + prices[i]。
- 冷冻期(dp[i][3]):
- 前一天卖出,进入冷冻期:dp[i-1][2]。
- dp[i][3] = dp[i-1][2]。
- 空间优化:
- 使用滚动数组 dp[2][4],通过 i % 2 交替存储当前和前一天状态,空间从 O(n) 优化到 O(1)。
- 状态转移只依赖前一天的 dp[i-1][j]。
- 时间复杂度:
- 遍历天数:O(n),n 为 prices.length。
- 每次更新4个状态:O(1)。
- 总时间复杂度:O(n)。
- 空间复杂度:
- 使用滚动数组 dp[2][4]:O(1)。
代码
class Solution {
/**
* 状态定义:
* dp[i][0]: 第i天持有股票的最大利润
* dp[i][1]: 第i天不持有股票且不在冷冻期(可买入)的最大利润
* dp[i][2]: 第i天卖出股票的最大利润
* dp[i][3]: 第i天处于冷冻期的最大利润
*/
public int maxProfit(int[] prices) {
int len = prices.length;
// 使用滚动数组优化空间,dp[2][4]交替存储当前和前一天的状态
int[][] dp = new int[2][4];
// 初始化第0天状态
dp[0][0] = -prices[0]; // 持有股票(买入)
// dp[0][1] = 0; // 不持有且不在冷冻期,默认0
// dp[0][2] = 0; // 卖出,默认0(不可达)
// dp[0][3] = 0; // 冷冻期,默认0(不可达)
// 遍历从第1天到最后一天
for (int i = 1; i < len; i++) {
// 持有股票:
// 1. 前一天持有:dp[i-1][0]
// 2. 前一天不持有且可买入,当天买入:dp[i-1][1] - prices[i]
// 3. 前一天冷冻期结束,当天买入:dp[i-1][3] - prices[i]
dp[i % 2][0] = Math.max(Math.max(dp[(i - 1) % 2][0], dp[(i - 1) % 2][1] - prices[i]),
dp[(i - 1) % 2][3] - prices[i]);
// 不持有且不在冷冻期:
// 1. 前一天不持有且可买入:dp[i-1][1]
// 2. 前一天冷冻期结束:dp[i-1][3]
dp[i % 2][1] = Math.max(dp[(i - 1) % 2][1], dp[(i - 1) % 2][3]);
// 当天卖出:
// 前一天持有,当天卖出:dp[i-1][0] + prices[i]
dp[i % 2][2] = dp[(i - 1) % 2][0] + prices[i];
// 冷冻期:
// 前一天卖出,进入冷冻期:dp[i-1][2]
dp[i % 2][3] = dp[(i - 1) % 2][2];
}
// 返回最后一天不持有股票的最大利润(可买入、卖出或冷冻期)
return Math.max(Math.max(dp[(len - 1) % 2][1], dp[(len - 1) % 2][2]), dp[(len - 1) % 2][3]);
}
}
更多推荐
所有评论(0)