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;
}
};