描述
有 n 个学生站成一排,每个学生有一个能力值,牛牛想从这 n 个学生中按照顺序选取 k 名学生,要求相邻两个学生的位置编号的差不超过 d,使得这 k 个学生的能力值的乘积最大,你能返回最大的乘积吗?
输入描述:
每个输入包含 1 个测试用例。每个测试数据的第一行包含一个整数 n (1 <= n <= 50),表示学生的个数,接下来的一行,包含 n 个整数,按顺序表示每个学生的能力值 ai(-50 <= ai <= 50)。接下来的一行包含两个整数,k 和 d (1 <= k <= 10, 1 <= d <= 50)。
输出描述:
输出一行表示最大的乘积。
public class Choir {
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
while (sc.hasNext()) {
int n = sc.nextInt();
int[] arr = new int[n+1];
for (int i = 1; i <= n; i++){ //从1开始
arr[i] = sc.nextInt();
}
int k = sc.nextInt();
int d = sc.nextInt();
long[][] Cmax = new long[k+1][n+1];//最大乘值
long[][] Cmin = new long[k+1][n+1];//最小乘值
long res = Integer.MIN_VALUE;
for (int i = 1;i <= n; i++){
Cmax[1][i] = arr[i];
Cmin[1][i] = arr[i];
for (int i1 =2; i1 <= k; i1++){
for (int j = i-1; j>0 && i-j <= d; j--){
Cmax[i1][i] = Math.max(Cmax[i1][i],Math.max(Cmax[i1-1][j] * arr[i],
Cmin[i1-1][j] * arr[i]));
Cmin[i1][i] = Math.min(Cmin[i1][i], Math.min(Cmax[i1-1][j] * arr[i],
Cmin[i1-1][j] * arr[i]));
}
}
res = Math.max(res, Cmax[k][i]);
}
System.out.println(res);
}
}
}