前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >LeetCode 0188 - Best Time to Buy and Sell Stock IV

LeetCode 0188 - Best Time to Buy and Sell Stock IV

作者头像
Reck Zhang
发布2021-08-11 14:55:00
2370
发布2021-08-11 14:55:00
举报
文章被收录于专栏:Reck ZhangReck ZhangReck Zhang

Best Time to Buy and Sell Stock IV

Desicription

Say you have an array for which the ith element is the price of a given stock on day i.

Design an algorithm to find the maximum profit. You may complete at most k transactions.

Note: You may not engage in multiple transactions at the same time (ie, you must sell the stock before you buy again).

Example 1:

Input: [2,4,1], k = 2
Output: 2
Explanation: Buy on day 1 (price = 2) and sell on day 2 (price = 4), profit = 4-2 = 2.

Example 2:

Input: [3,2,6,5,0,3], k = 2
Output: 7
Explanation: Buy on day 2 (price = 2) and sell on day 3 (price = 6), profit = 6-2 = 4.
             Then buy on day 5 (price = 0) and sell on day 6 (price = 3), profit = 3-0 = 3.

Solution

class Solution {
public:
    int maxProfit(int k, vector<int>& prices) {
        priority_queue<int> profits;
        stack<pair<int, int>> vps;
        int v = 0;
        int p = 0;
        int len = prices.size();
        while(p < len) {
            for(v = p; v < len-1 && prices[v] >= prices[v+1]; v++);
            for(p = v+1; p < len && prices[p] >= prices[p-1]; p++);
            while(vps.size() && prices[vps.top().first] > prices[v]) {
                profits.push(prices[vps.top().second - 1] - prices[vps.top().first]);
                vps.pop();
            }
            while(vps.size() && prices[vps.top().second-1] <= prices[p-1]) {
                profits.push(prices[vps.top().second-1] - prices[v]);
                v = vps.top().first;
                vps.pop();
            }
            vps.push(pair<int, int>(v, p));
        }
        while(vps.size()) {
            profits.push(prices[vps.top().second-1] - prices[vps.top().first]);
            vps.pop();
        }
        int res = 0;
        while(k-- && profits.size())
            res += profits.top(), profits.pop();
        return res;
    }
};
本文参与 腾讯云自媒体分享计划,分享自作者个人站点/博客。
原始发表:2018-06-06,如有侵权请联系 cloudcommunity@tencent.com 删除

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

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

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

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • Best Time to Buy and Sell Stock IV
    • Desicription
      • Solution
      领券
      问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档