当前位置:   article > 正文

LeetCode 188题: 买卖股票的最佳时机IV优化(原创)

LeetCode 188题: 买卖股票的最佳时机IV优化(原创)

之前完成了LeetCode 188题: 买卖股票的最佳时机IV(原创)-CSDN博客,虽然完成代码编写,并提交成功,但运行效率还未达到最优的1ms,见下图:

仔细检查代码,感觉还是有可优化的点:

  1.  由于只需要返回最后一天的最大获利值,因此之前所有天数的最大获利值不需要存储;
  2. 不能连续够买,也意味着没法连续卖,因为手头上最多只有一只股票;
  3. 必须先买才能后卖,因此第n次卖的操作是基于第n次买的操作基础上;而第n次买的操作是基于第n-1次卖的操作,
  4. 另外,第一次买的的操作属于特殊情况,因为没有上一次卖的操作,获利值等于-prices[0]

按照这个思路完成代码修改,修改之后的代码如下:

  1. package dp;
  2. public class MaxProfit4 {
  3. public static void main(String[] args) {
  4. int[] prices = {3, 2, 6, 5, 0, 3};
  5. int k = 2;
  6. int result = new MaxProfit4().maxProfit(k, prices);
  7. System.out.println("计算结果:" + result);
  8. }
  9. public int maxProfit(int k, int[] prices) {
  10. int[] buys = new int[k];
  11. int[] sells = new int[k];
  12. for (int i = 0; i < k; i++) {
  13. buys[i] = -prices[0];
  14. }
  15. for (int i = 1; i < prices.length; i++) {
  16. for (int j = 0; j < k; j++) {
  17. // j-1天卖掉的获利值,第1天为0
  18. int sellJMinus1 = j > 0 ? sells[j - 1] : 0;
  19. // 买了j次的最大获利值等于:昨天买了j次,和昨天卖了j-1次,今天再买的最大值
  20. buys[j] = Math.max(buys[j], sellJMinus1 - prices[i]);
  21. // 卖了j次的最大获利值等于:昨天卖了j次,和昨天买了j次,今天再卖掉的最大值
  22. sells[j] = Math.max(sells[j], buys[j] + prices[i]);
  23. }
  24. }
  25. return sells[k - 1];
  26. }
  27. }

提交之后果然性能优化到1ms,见下图:

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/我家自动化/article/detail/864853
推荐阅读
相关标签
  

闽ICP备14008679号