前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >LeetCode 188. 买卖股票的最佳时机 IV(动态规划)

LeetCode 188. 买卖股票的最佳时机 IV(动态规划)

作者头像
Michael阿明
发布2020-07-13 12:01:42
6240
发布2020-07-13 12:01:42
举报

1. 题目

给定一个数组,它的第 i 个元素是一支给定的股票在第 i 天的价格。

设计一个算法来计算你所能获取的最大利润。你最多可以完成 k 笔交易。

注意: 你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。

代码语言:javascript
复制
示例 1:
输入: [2,4,1], k = 2
输出: 2
解释: 在第 1 天 (股票价格 = 2) 的时候买入,
在第 2 天 (股票价格 = 4) 的时候卖出,这笔交易所能获得利润 = 4-2 = 2 。

示例 2:
输入: [3,2,6,5,0,3], k = 2
输出: 7
解释: 在第 2 天 (股票价格 = 2) 的时候买入,
在第 3 天 (股票价格 = 6) 的时候卖出, 这笔交易所能获得利润 = 6-2 = 4 。
     随后,在第 5 天 (股票价格 = 0) 的时候买入,
     在第 6 天 (股票价格 = 3) 的时候卖出, 这笔交易所能获得利润 = 3-0 = 3 。

来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/best-time-to-buy-and-sell-stock-iv 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

2. 解题

这道题需要考虑一个 k 非常大的时候,k > n/2 时,就可以看做股票第2题了,无限次数。最多可以交易 min(k, n/2) 次

其他情况跟第3题一样

  • dp[i][k][0]表示第 i 天,经过最多 k 次交易,0不持有股票的最大收益
  • dp[i][k][1]表示第 i 天,经过最多 k 次交易,1持有股票的最大收益
代码语言:javascript
复制
class Solution {
public:
    int maxProfit(int k, vector<int>& prices) {
        if(prices.empty())
            return 0;
        int i, deal, n = prices.size();
        if(k > n/2)//股票第二题
        {
            int maxprofit = 0;
            for(int i = 1; i < prices.size(); i++) 
            {
                if(prices[i] > prices[i-1])
                    maxprofit += prices[i] - prices[i-1];
            }
            return maxprofit;
        }
        int dp[n][k+1][2];
        memset(dp,0,sizeof(dp));
        for(i = 0; i < n; ++i)
        {
            for(deal = 1; deal <= k; ++deal)
            {
                if(i == 0)
                {
                    dp[i][deal][0] = 0;
                    dp[i][deal][1] = -prices[0];
                    continue;
                }
                dp[i][deal][0] = max(dp[i-1][deal][0], dp[i-1][deal][1]+prices[i]);
                dp[i][deal][1] = max(dp[i-1][deal][1], dp[i-1][deal-1][0]-prices[i]);
            }
        }
        return dp[n-1][k][0];
    }
};

16 ms 12.8 MB

本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2020/04/30 ,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 1. 题目
  • 2. 解题
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档