有 n 个学生站成一排,每个学生有一个能力值,牛牛想从这 n 个学生中按照顺序选取 k 名学生,要求相邻两个学生的位置编号的差不超过 d,使得这 k 个学生的能力值的乘积最大,你能返回最大的乘积吗?
输入描述:
每个输入包含 1 个测试用例。每个测试数据的第一行包含一个整数 n (1 <= n <= 50),表示学生的个数,接下来的一行,包含 n 个整数,按顺序表示每个学生的能力值 ai(-50 <= ai <= 50)。接下来的一行包含两个整数,k 和 d (1 <= k <= 10, 1 <= d <= 50)。
输出描述:
输出一行表示最大的乘积。
输入例子:
3
7 4 7
2 50
输出例子:
49
解题思路:动态规划
import java.util.Arrays;
import java.util.Scanner;
public class WangyiTest8 {
public static void main(String[] args) {
Scanner scanner = new Scanner(System.in);
while (scanner.hasNext()) {
int n = scanner.nextInt();
int[] a = new int[n + 1];
for (int i = 1; i <= n; i++) {
a[i] = scanner.nextInt();
}
int k = scanner.nextInt();
int d = scanner.nextInt();
System.out.println(getMaxPower(n, a, k, d));
}
}
public static long getMaxPower(int n, int[] a, int k, int d) {
long[][] dmax = new long[k + 1][n + 1];// 记录最大乘积
long[][] dmin = new long[k + 1][n + 1];// 记录最大乘积
// dmax[k][i]表示 : 当选中了k个学生,并且以第i个学生为结尾,所产生的最大乘积;
// dmin[k][i]表示 : 当选中了k个学生,并且以第i个学生为结尾,所产生的最小乘积;
long maxPower = Integer.MIN_VALUE;
for (int i = 1; i <= n; i++) {
//// 初始化 选取了一个学生后
dmax[1][i] = a[i];
dmin[1][i] = a[i];
for (int stuNum = 2; stuNum <= k; stuNum++) {
for (int j = i - 1; j > 0 && i - j <= d; j--) {
// 选择下一个学生的时候,有可能是负数乘以最小的值变成了最大的数。因为负负得正。
dmax[stuNum][i] = Math.max(dmax[stuNum][i],
Math.max(dmax[stuNum - 1][j] * a[i], dmin[stuNum - 1][j] * a[i]));
dmin[stuNum][i] = Math.min(dmin[stuNum][i],
Math.min(dmin[stuNum - 1][j] * a[i], dmax[stuNum - 1][j] * a[i]));
}
}
maxPower = Math.max(dmax[k][i], maxPower);
}
return maxPower;
}
}