赞
踩
之前完成了LeetCode 188题: 买卖股票的最佳时机IV(原创)-CSDN博客,虽然完成代码编写,并提交成功,但运行效率还未达到最优的1ms,见下图:
仔细检查代码,感觉还是有可优化的点:
按照这个思路完成代码修改,修改之后的代码如下:
- package dp;
-
- public class MaxProfit4 {
-
- public static void main(String[] args) {
- int[] prices = {3, 2, 6, 5, 0, 3};
- int k = 2;
- int result = new MaxProfit4().maxProfit(k, prices);
- System.out.println("计算结果:" + result);
- }
-
- public int maxProfit(int k, int[] prices) {
- int[] buys = new int[k];
- int[] sells = new int[k];
- for (int i = 0; i < k; i++) {
- buys[i] = -prices[0];
- }
- for (int i = 1; i < prices.length; i++) {
- for (int j = 0; j < k; j++) {
- // j-1天卖掉的获利值,第1天为0
- int sellJMinus1 = j > 0 ? sells[j - 1] : 0;
- // 买了j次的最大获利值等于:昨天买了j次,和昨天卖了j-1次,今天再买的最大值
- buys[j] = Math.max(buys[j], sellJMinus1 - prices[i]);
- // 卖了j次的最大获利值等于:昨天卖了j次,和昨天买了j次,今天再卖掉的最大值
- sells[j] = Math.max(sells[j], buys[j] + prices[i]);
- }
- }
- return sells[k - 1];
- }
- }

提交之后果然性能优化到1ms,见下图:
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。