合唱团

题目描述
有 n 个学生站成一排，每个学生有一个能力值，牛牛想从这 n 个学生中按照顺序选取 k 名学生，要求相邻两个学生的位置编号的差不超过 d，使得这 k 个学生的能力值的乘积最大，你能返回最大的乘积吗？
输入描述:
每个输入包含 1 个测试用例。每个测试数据的第一行包含一个整数 n (1 <= n <= 50)，表示学生的个数，接下来的一行，包含 n 个整数，按顺序表示每个学生的能力值 ai（-50 <= ai <= 50）。接下来的一行包含两个整数，k 和 d (1 <= k <= 10, 1 <= d <= 50)。
输出描述:
输出一行表示最大的乘积。

import java.util.Arrays;
import java.util.Scanner;

public class Main {

    public static void main(String[] args) {
        Scanner scan=new Scanner(System.in);
        int n=scan.nextInt();
        int []arr=new int[n];
        for(int i=0;i<n;i++){
            arr[i]=scan.nextInt();
        }
        int k=scan.nextInt();
        int d=scan.nextInt();
        long []dp=new long[n];
        long []dp2=new long[n];
        for(int j=0;j<k;j++){
            for(int i=n-1;i>=j;i--){
                if(j==0){
                    dp[i]=arr[i];
                    dp2[i]=arr[i];
                    continue;
                }
                long max=Long.MIN_VALUE;
                long min=Long.MAX_VALUE;
                for(int m=i-1;m>=j-1&&m+d>=i;m--){
                   if(dp[m]>max){
                       max=dp[m];
                   }
                    if(dp2[m]<min){
                        min=dp2[m];
                    }
                }
                if(arr[i]>=0){
                    dp[i]=max*arr[i];
                    dp2[i]=min*arr[i];
                }else{
                    dp2[i]=max*arr[i];
                    dp[i]=min*arr[i];
                }
            }
        }
        long max=Integer.MIN_VALUE;
        for(int i=k-1;i<n;i++){
            if(dp[i]>max){
                max=dp[i];
            }
        }
        System.out.println(max);
    }
}