题目链接
1057. 股票买卖 IV - AcWing题库高质量的算法题库https://www.acwing.com/problem/content/1059/今天是个状态机的第二讲,有了昨天和今天两道题的铺垫,能够对状态机有初步的认识。
分析题目,在长度为n的数组中,最多k对数的后序减先序之差求和取最大值,任意一对数,只能同时小于或者大于其他对数。对于此题,我们则使用f[N][M][2]容器进行状态转移,将f[ i ][ j ][0/1]视为,从第1个物品到第i个物品,交易了j次,手上有没有股票(有票=1,无票=0),如果手上有票,可以从f[ i-1][ j ][1]:前一项有票同一交易次数转移过来或者从f[ i-1][ j ][0] - w[ i ]:前一项无票同一交易次数减去当前股票的价格转移过来(买入操作不改变交易次数,卖出改变交易次数),取二者最大,反之同理
f[i][j][0]=max(f[i-1][j][0],f[i-1][j][1]+w[i]);
f[i][j][1]=max(f[i-1][j][1],f[i-1][j-1][0]-w[i]);
ac代码为
#include<bits/stdc++.h>
using namespace std;
const int N=100010;
const int M=110;
int f[N][M][2];
int n,m;
int w[N];
int main(){
cin>>n>>m;
memset(f,-0x3f,sizeof f);
for(int i=1;i<=n;i++){
cin>>w[i];
f[i][0][0]=0;
}
f[0][0][0]=0;
for(int i=1;i<=n;i++){
for(int j=1;j<=m;j++){
f[i][j][0]=max(f[i-1][j][0],f[i-1][j][1]+w[i]);
f[i][j][1]=max(f[i-1][j][1],f[i-1][j-1][0]-w[i]);
}
}
int res=0;
for(int i=1;i<=m;i++)res=max(res,f[n][i][0]);
cout<<res;
}