【acwing】1057. 股票买卖 IV**(状态机模型|DP)

穿越隧道

#include <bits/stdc++.h>
using namespace std;
const int N = 1e5 + 10, M = 1e2 + 10;
int n,k;
int a[N];
int f[N][M][2];
int main(){
	scanf("%d%d",&n,&k);
	for(int i = 1; i <= n; i++){
		scanf("%d",&a[i]);
	}
	memset(f,-0x3f,sizeof(f));
// 	f[0][0][0] = 0;
	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][0],f[i - 1][j ][1] + a[i]);
			f[i][j][1] = max(f[i - 1][j][1],f[i - 1][j - 1][0] - a[i]);
		}
	} 
	int res = 0;
	for(int i = 0; i <= k; i++){//可能0次交易、1次交易、2次交易就可能达到收益最大 
		res = max(res,f[n][i][0]);
	}
	printf("%d\n",res);
	return 0;
}