写点什么

代码随想录 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
用户头像

jjn0703

关注

Java工程师/终身学习者 2018-03-26 加入

USTC硕士/健身健美爱好者/Java工程师.

评论

发布
暂无评论
代码随想录 Day51 - 动态规划(十二)_jjn0703_InfoQ写作社区