#include <stdio.h> #include <algorithm> #include <string.h> using namespace std; #define oo 0x3f3f3f3f int P[1005], dp[1005][1005]; int main() { int T; scanf("%d", &T); while(T--) { int n, k; scanf("%d %d", &n, &k); for(int i=1; i<=n; i++) scanf("%d", &P[i]); int maxn; memset(dp, 0, sizeof(dp)); for(int i=1; i<=k; i++) { maxn = -P[1]; for(int j=2; j<=n; j++) { maxn = max(maxn, dp[i-1][j-1]-P[j]); dp[i][j] = max(dp[i][j-1], maxn+P[j]); } } printf("%d ", dp[k][n]); } return 0; }
dp[i][j]表示第i次交易第j天的股票时可获得的最大利益。第i次交易的最大利益为此次不进行交易,或者进行交易也就是
上次交易的于某天的利益(此天在j之前)和此天后再次买入卖出获得的利益两者之和,遍历所有可能状态即可,
为了防止超时在查找j天之前的最大利益时加一变量maxn记录在j天之前的最大价值的交易,省去了一重循环。