注意以下几个初始化:
- 刚开始的不可达状态要全部设置成负无穷(即刚开始买过股票的状态)
- 对于每一个i,fi01要初始化为0
- f001也要初始化为0
还要注意。最终状态在fin1中遍历,一定要是完整交易,要卖掉
#include <iostream>
#include <cstring>
using namespace std;
const int N=100010;
int a[N];
int f[N][110][2];
int main()
{
int n,m;
cin>>n>>m;
memset(f,-0x3f,sizeof f);
for(int i=1;i<=n;i++)
{
cin>>a[i];
f[i][0][1]=0;
}
f[0][0][1]=0;
for(int i=1;i<=n;i++)
{
for(int j=1;j<=m;j++)
{
//0是买过的状态,1是卖过的状态
//当讨论fij时,看f上一回合(即i-1天)的状态怎么转移到i的
//只有买股票的时候要花费一次操作,卖股票不算
f[i][j][1]=max(f[i-1][j][1],f[i-1][j][0]+a[i]);
f[i][j][0]=max(f[i-1][j][0],f[i-1][j-1][1]-a[i]);
}
}
int res=0;
for(int i=0;i<=m;i++)
{
res=max(res,f[n][i][1]);
}
cout<<res;
return 0;
}
2023/11/28
fijk: 在前i个股票中买,交易数不超过j,状态为k
还是要注意状态边界,即入口的问题:
1. 刚开始的不可达状态要全部设置成负无穷(即刚开始买过股票的状态)
2. i要从0开始初始化,不能从1开始:
// 状态机入口:最开始手里一定没有股
memset(f, -0x3f, sizeof(f));
for(int i=0;i<=n;i++)
{
f[i][0][0] = 0;
}
#include <iostream>
#include <cstring>
using namespace std;
const int N=100010;
// fijk: 在前i个股票中买,交易数不超过j,状态为k
int f[N][110][2];
int a[N];
int n,k;
int main()
{
cin>>n>>k;
for(int i=1;i<=n;i++)
{
cin>>a[i];
}
// 状态机入口:最开始手里一定没有股
memset(f, -0x3f, sizeof(f));
for(int i=0;i<=n;i++)
{
f[i][0][0] = 0;
}
for(int i=1;i<=n;i++)
{
for(int j=1;j<=k;j++)
{
// 手中无股的状态
f[i][j][0] = max(f[i-1][j][1]+a[i], f[i-1][j][0]);
// 手中有股的状态
f[i][j][1] = max(f[i-1][j-1][0]-a[i], f[i-1][j][1]);
}
}
int res = 0;
// 最终不一定非要进行k次交易
for(int i=1;i<=k;i++)
{
res = max(res, f[n][i][0]);
}
cout<<res;
return 0;
}