leetcode 188. 买卖股票的最佳时机 IV--zhengjun

先用 \(f_{i,j}\) 表示第 \(i\) 次交易,这次交易到 \(j\) 的最大利润。

这样 \(f_i\) 就可以从 \(f_{i-1}\) 那里直接得到,(参见leetcode123. 买卖股票的最佳时机 III--zhengjun

还要注意一下空间,要滚动

代码

class Solution {
public:
    int n,f[2][10001];
    int maxProfit(int k,vector<int>& a) {
        int now=1,last=0;
        n=a.size();
        if(k>n/2)k=n/2;
        int ans=0;
        for(int i=1;i<=k;i++){
            memset(f[now],0,sizeof(f[now]));
            int maxx=-0x3fffffff;
            for(int j=0;j<n;j++){
                f[now][j]=max(maxx+a[j],j?f[now][j-1]:0);
                if(maxx<f[last][j]-a[j])maxx=f[last][j]-a[j];
            }
            if(ans<f[now][n-1])ans=f[now][n-1];
            now^=1;last^=1;
        }
        return ans;
    }
};
posted @ 2022-06-11 15:09  A_zjzj  阅读(20)  评论(0编辑  收藏  举报