成魔之路−> 算法提高课题解
思路:
-
1.状态表示
集合:从前 i 只股票中选,正在进行第 j 次交易,f[i][j][0] 表示手中无货,f[i][j][1] 表示手中有货
属性:max -
2. 状态转移
手中无货\Rightarrow 手中无货:f[i][j][0]=f[i-1][j][0]
手中有货\Rightarrow 手中无货:f[i][j][0]=f[i-1][j][1]+w[i]
因此,手中无货:f[i][j][0]=\max(f[i-1][j][0],f[i-1][j][1]+w[i])
手中有货\Rightarrow 手中有货:f[i][j][1]=f[i-1][j][1]
手中无货\Rightarrow 手中有货:f[i][j][1]=f[i-1][j-1][0]-w[i]
因此,手中有货:f[i][j][1]=\max(f[i-1][j][1],f[i-1][j-1][0]-w[i]) -
3. 优化
手中无货:f[j][0]=\max(f[j][0],f[j][1]+w[i])
手中有货:f[j][1]=\max(f[j][1],f[j-1][0]-w[i]) -
4. 提醒
一次完整的买入卖出才算一次交易的真正结束
完整代码
#include<bits/stdc++.h>
using namespace std;
const int N = 100010, M = 110;
int n,m;
int w[N];
int f[M][2];
int main()
{
cin>>n>>m;
for(int i=1;i<=n;i++) cin>>w[i];
memset(f,-0x3f,sizeof f);
f[0][0]=0;
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
{
f[j][0]=max(f[j][0],f[j][1]+w[i]); //手中无货
f[j][1]=max(f[j][1],f[j-1][0]-w[i]); //手中有货
}
int res=0;
for(int i=0;i<=m;i++) res=max(res,f[i][0]);
cout<<res<<endl;
return 0;
}