【状态机DP】股票买卖 IV

题目

题目来源：acwing

给定一个长度为 N 的数组，数组中的第 i 个数字表示一个给定股票在第 i 天的价格。

设计一个算法来计算你所能获取的最大利润，你最多可以完成 k 笔交易。

注意：你不能同时参与多笔交易（你必须在再次购买前出售掉之前的股票）。一次买入卖出合为一笔交易。

输入格式
第一行包含整数 N 和 k，表示数组的长度以及你可以完成的最大交易数量。

第二行包含 N 个不超过 10000 的正整数，表示完整的数组。

输出格式
输出一个整数，表示最大利润。

数据范围
1≤N≤105,
1≤k≤100
输入样例1：
3 2
2 4 1
输出样例1：
2
输入样例2：
6 2
3 2 6 5 0 3
输出样例2：
7
样例解释
样例1：在第 1 天 (股票价格 = 2) 的时候买入，在第 2 天 (股票价格 = 4) 的时候卖出，这笔交易所能获得利润 = 4-2 = 2 。

样例2：在第 2 天 (股票价格 = 2) 的时候买入，在第 3 天 (股票价格 = 6) 的时候卖出, 这笔交易所能获得利润 = 6-2 = 4 。随后，在第 5 天 (股票价格 = 0) 的时候买入，在第 6 天 (股票价格 = 3) 的时候卖出, 这笔交易所能获得利润 = 3-0 = 3 。共计利润 4+3 = 7.

题目思路

这个题目抓住一个关键点，就是手中只能持有一个股票。
我们对这个进行分析可以发现，手中的状态可以分为：手中无货，手中有货。
当手中有货的时候，可以第二天卖掉，变成手中无货状态，也可以不卖掉，继续保持手中有货的状态。
当手中无货的时候，可以第二天买进， 变成有货状态，也可以不买，继续保持手中无货的状态。

根据这个分析，我们可以发现，这是一个状态机DP。

他的集合表示为：从i个货中选，且交益数量不超过j状态为 0/1的利润值。
状态属性： max

状态转移方程为：
$f_{i,j,0}=max(f_{i-1,j,1}+w,f_{i-1,j,0})$
$f_{i,j,1}=max(f_{i-1,j-1,1}-w,f_{i-1,j,1})$

AC代码

#include<bits/stdc++.h>
using namespace std;

#define _for(i, a, b) for (int i = (a); i < (b); ++i)
#define _rep(i, a, b) for (int i = (a); i <= (b); ++i)
#define For(i, a, b) for (int i = (a); i >= (b); --i)
#define debug(a) cout << #a << " = " << a << ENDL
#define mod(x) (x) % MOD
#define ENDL "\n"
typedef long long ll;
typedef pair<int, int> pii;
typedef vector<int> vi;

const int M = 100000 + 5, N = 100 + 5, INF = 0x3f3f3f3f;
int f[M][N][2];

int main() {
#ifdef LOCAL
	freopen("data.in", "r", stdin);
#endif // LOCAL
	ios::sync_with_stdio(false);
	cin.tie(0), cout.tie(0);

	int n, m;
	cin >> n >> m;

	memset(f, -0x3f, sizeof f);
	_rep(i, 0, n) f[i][0][0] = 0;

	_rep(i, 1, n) {
		int w;
		cin >> w;
		_rep(j, 1, m) {
			f[i][j][0] = max(f[i - 1][j][1] + w, f[i - 1][j][0]);
			f[i][j][1] = max(f[i - 1][j - 1][0] - w, f[i - 1][j][1]);
		}
	}

	int ans = 0;
	_rep(i, 1, m) ans = max(ans, f[n][i][0]);
	cout << ans << ENDL;
	return 0;
}

反思

1. 想问题没有考虑周全，不能一次AC，自己没有去想额外的数据点，这个毛病要改掉。
2. 初始化状态没有思考清楚，太浮躁了。以后做题思考，要按照之前的规定的做。
3. 数组也没有仔细看，要不就是开大了炸了，要不就是开小了WA了。
4. 状态转移也需要考虑周全，不能少掉任何一种情况。