LeetCode 188. Best Time to Buy and Sell Stock IV

最新推荐文章于 2023-02-22 06:47:24 发布

dongcl97

最新推荐文章于 2023-02-22 06:47:24 发布

阅读量92

点赞数

分类专栏： LeetCode 算法文章标签： leetcode 算法

本文链接：https://blog.csdn.net/weixin_36634831/article/details/105177294

版权

LeetCode 同时被 2 个专栏收录

58 篇文章 0 订阅

订阅专栏

算法

29 篇文章 0 订阅

订阅专栏

Problem

题目链接

Notes

买卖股票系列第四题，经过买卖股票第三题的一顿操作之后，其实我们可以得出买卖k次股票的通用解法，但是会有k非常大的测试样例，所以即使我们在第三题里把dp数组降到了长度为k的数组也会爆内存。但仔细一想，因为题目规定买卖不能在同一天，如果k>=n/2，那么买卖k次和买卖无限次是一样的，那就是和该系列的第二题解法一致，而第二题可以用空间复杂度为o(1）的方法解决。那么该问题也就解决了。

Codes

class Solution {
public:
    int maxProfit_infinite(vector<int>& prices) {
        int n=prices.size();
        if(n<=1) return 0;
        int ans=0;
        int nowprice=prices[0];
        for(int i=1;i<n;++i)
        {
            if(prices[i]<prices[i-1])
            {
                ans+=prices[i-1]-nowprice;
                nowprice=prices[i];
            }
        }
        if(prices[n-1]>=prices[n-2])
            ans+=prices[n-1]-nowprice;
        return ans;
    }
    int maxProfit(int k, vector<int>& prices) {
        int n=prices.size();
        if(n<=1||k<=0) return 0;
        if(k>=n/2) return maxProfit_infinite(prices);
        int dp[k+1];
        int m[k+1];
        for(int i=0;i<=k;++i) dp[i]=0;
        for(int i=1;i<=k;++i) m[i]=prices[0];
        for(int i=1;i<n;++i)
        {
            for(int j=1;j<=k;++j)
            {
                dp[j]=max(dp[j],prices[i]-m[j]);
                m[j]=min(m[j],prices[i]-dp[j-1]);
            }
        }
        return dp[k];
    }
};