import java.util.*; public class Solution { /** * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可 * * * @param prices int整型一维数组 * @return int整型 */ public int maxProfit (int[] prices) { // write code here int n = prices.length; int[][][] f = new int[n + 1][4][2]; for (int i = 0; i <= 3; ++i) { f[0][i][1] = -1000000; } for (int i = 1; i <= n; ++i) { for (int j = 1; j <= 3; ++j) { f[i][j][0] = Math.max(f[i - 1][j][0], f[i - 1][j][1] + prices[i - 1]); f[i][j][1] = Math.max(f[i - 1][j][1], f[i - 1][j - 1][0] - prices[i - 1]); } } return f[n][3][0]; } }
编程语言是Java。
该题考察的知识点包括:
- 动态规划
- 多维数组操作
代码的文字解释:
- 获取数组的长度:获取数组
prices
的长度,即价格的天数n
。 - 创建状态数组:使用
int[][][] f
来表示每天的状态以及每种交易次数下的最大利润。数组大小为(n + 1) x 4 x 2
。 - 初始化状态数组:将数组
f
的初始值设置为特定的数值,根据题意为-1e6
(一个较小的负数)。 - 动态规划状态转移:通过两层嵌套的循环遍历每天和交易次数的状态,根据状态转移方程更新状态数组
f
。其中,f[i][j][0]
表示第i
天、已进行j
次交易、不持有牛群时的最大利润;f[i][j][1]
表示第i
天、已进行j
次交易、持有牛群时的最大利润。 - 返回结果:返回
f[n][3][0]
,表示最后一天、已进行 3 次交易、不持有牛群时的最大利润,即题目要求的答案。