1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33
| #include <vector> #include <algorithm> using std::vector;
class Solution { public: int maxProfit(int k, vector<int> &prices) { int i, j, n = prices.size(); int nr_state = 2 * k + 1; if (n == 1) { return 0; } vector<vector<int> > dp(n, vector<int>(nr_state, 0)); dp[0][0] = 0; dp[0][1] = -prices[0]; for (j = 2; j < nr_state; j++) { dp[i][j] = -1e9; }
for (i = 1; i < n; i++) { dp[i][0] = dp[i - 1][0]; for (j = 1; j < nr_state; j += 2) { dp[i][j] = std::max(dp[i - 1][j], dp[i - 1][j - 1] - prices[i]); } for (j = 2; j < nr_state; j += 2) { dp[i][j] = std::max(dp[i - 1][j], dp[i - 1][j - 1] + prices[i]); } } return *std::max_element(dp[n - 1].begin(), dp[n - 1].end()); } };
|