代码随想录 Day51 - 动态规划(十二)
作者:jjn0703
- 2023-08-24 江苏
本文字数:1321 字
阅读完需:约 4 分钟
作业题
309. 买卖股票的最佳时机含冷冻期
链接:https://leetcode.cn/problems/best-time-to-buy-and-sell-stock-with-cooldown/description/
package jjn.daily;
import java.util.Scanner;
/**
* @author Jjn
* @since 2023/8/24 23:31
*/
public class LeetCode309 {
public int maxProfit(int[] prices) {
// 状态包含:持股/卖出且冷冻/卖出非冷冻
int[][] dp = new int[prices.length][3];
dp[0][0] = -prices[0];
dp[0][1] = 0;
dp[0][2] = 0;
for (int i = 1; i < prices.length; i++) {
// dp[i][0] - 持有股票
// 1.保留昨天买的股票
// 2. 卖出非冷冻期的股票
dp[i][0] = Math.max(dp[i - 1][0], dp[i - 1][2] - prices[i]);
// dp[i][1]
// 卖出昨天购买的股票
dp[i][1] = dp[i - 1][0] + prices[i];
// dp[i][2]
// 1.昨天卖出,非冷冻
// 2. 昨天没卖出,非冷冻
dp[i][2] = Math.max(dp[i - 1][1], dp[i - 1][2]);
}
// 若最后一天依然持有股票,显然不是最佳解
return Math.max(dp[prices.length - 1][1], dp[prices.length - 1][2]);
}
public static void main(String[] args) {
Scanner scanner = new Scanner(System.in);
int total = scanner.nextInt();
int[] prices = new int[total];
for (int i = 0; i < total; i++) {
prices[i] = scanner.nextInt();
}
int maxProfit = new LeetCode309().maxProfit(prices);
System.out.println(maxProfit);
}
}
复制代码
714. 买卖股票的最佳时机含手续费
链接:https://leetcode.cn/problems/best-time-to-buy-and-sell-stock-with-transaction-fee/description/
package jjn.daily;
import java.util.Scanner;
/**
* @author Jjn
* @since 2023/8/24 23:48
*/
public class LeetCode714 {
public int maxProfit(int[] prices, int fee) {
int[][] profit = new int[prices.length][2];
profit[0][0] = 0;
profit[0][1] = -prices[0];
for (int i = 1; i < prices.length; i++) {
// 已经持有股票,卖出去
profit[i][0] = Math.max(profit[i - 1][0], profit[i - 1][1] + prices[i] - fee);
profit[i][1] = Math.max(profit[i - 1][1], profit[i - 1][0] - prices[i]);
}
return profit[prices.length - 1][0];
}
public static void main(String[] args) {
Scanner scanner = new Scanner(System.in);
int fee = scanner.nextInt();
int total = scanner.nextInt();
int[] prices = new int[total];
for (int i = 0; i < total; i++) {
prices[i] = scanner.nextInt();
}
int maxProfit = new LeetCode714().maxProfit(prices, fee);
System.out.println(maxProfit);
}
}
复制代码
划线
评论
复制
发布于: 2023-08-24阅读数: 11
版权声明: 本文为 InfoQ 作者【jjn0703】的原创文章。
原文链接:【http://xie.infoq.cn/article/e8040703c6608afc039530bca】。
本文遵守【CC-BY 4.0】协议,转载请保留原文出处及本版权声明。
jjn0703
关注
Java工程师/终身学习者 2018-03-26 加入
USTC硕士/健身健美爱好者/Java工程师.
评论