网易校招题目 合唱团 Java代码

题目描述

有 n 个学生站成一排，每个学生有一个能力值，牛牛想从这 n 个学生中按照顺序选取 k 名学生，要求相邻两个学生的位置编号的差不超过 d，使得这 k 个学生的能力值的乘积最大，你能返回最大的乘积吗？

输入描述:

每个输入包含 1 个测试用例。每个测试数据的第一行包含一个整数 n (1 <= n <= 50)，表示学生的个数，接下来的一行，包含 n 个整数，按顺序表示每个学生的能力值 ai（-50 <= ai <= 50）。接下来的一行包含两个整数，k 和 d (1 <= k <= 10, 1 <= d <= 50)。

输出描述:

输出一行表示最大的乘积。

示例1

输入

复制

3
7 4 7
2 50

输出

复制

49

import java.util.Scanner;

public class Main {
	public static long result=Long.MIN_VALUE;
	public static void main(String args[]){
		Scanner aa=new Scanner(System.in);
		int n=aa.nextInt();
		int ability[]=new int[n];
		for(int i=0;i<n;i++){
			ability[i]=aa.nextInt();
		}
		int k=aa.nextInt();
		int d=aa.nextInt();
		long dpmax[][]=new long[n][k+1];
		long dpmin[][]=new long[n][k+1];
		for(int i=0;i<n;i++){
			dpmax[i][1]=ability[i];
			dpmin[i][1]=ability[i];
		}
		for(int i=0;i<n;i++){
			for(int j=2;j<=k;j++){
				for(int r=Math.max(0, i-d);r<i;r++){
					dpmax[i][j]=Math.max(dpmax[i][j], Math.max(dpmax[r][j-1]*ability[i], dpmin[r][j-1]*ability[i]));
					dpmin[i][j]=Math.min(dpmin[i][j], Math.min(dpmax[r][j-1]*ability[i], dpmin[r][j-1]*ability[i]));
				}
			}
		}
		for(int i=0;i<n;i++){
			result=Math.max(result, Math.max(dpmax[i][k],dpmin[i][k]));
		}
		System.out.println(result);
		
	}
	
}