合唱团

题目描述

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

输入描述:

每个输入包含 1 个测试用例。每个测试数据的第一行包含一个整数 n (1 <= n <= 50)，表示学生的个数，接下来的一行，包含 n 个整数，按顺序表示每个学生的能力值 a

_i

（-50 <= ai <= 50）。接下来的一行包含两个整数，k 和 d (1 <= k <= 10, 1 <= d <= 50)。

import java.io.BufferedReader;
import java.io.InputStreamReader;

public class Main{
    public static void main(String[] args) throws Exception {
        BufferedReader read=new BufferedReader(new InputStreamReader(System.in));
        int n=Integer.parseInt(read.readLine());
        int [] nums=new int[n];
        String []xz=read.readLine().split(" ");
        for(int i=0;i<n ; i++){
          nums[i]=Integer.parseInt(xz[i]);
        }
        String []dsl=read.readLine().split(" ");
        int k = Integer.parseInt(dsl[0]);
        int d=Integer.parseInt(dsl[1]);
        long [][]min=new long[k][n];
        long [][]max=new long[k][n];
        long fs=0;
        for(int i =0; i<k;i++){
            for(int j=0;j<n;j++){
                if(i==0){
                    min[0][j]=nums[j];
                    max[0][j]=nums[j];
                }else{
                    min[i][j]=1;
                    max[i][j]=1;
                }
                for(int m=1;m<=d;m++){
                    if(j>=m){
                        if(i>0){
                            if(nums[j]>0){
                                min[i][j]=Math.min(min[i][j],min[i-1][j-m]*nums[j]);
                                max[i][j]=Math.max(max[i][j], max[i-1][j-m]*nums[j]);
                            }else{
                                min[i][j]=Math.min(min[i][j],max[i-1][j-m]*nums[j]);
                                max[i][j]=Math.max(max[i][j], min[i-1][j-m]*nums[j]);
                            }
                        }
                    }
                    if(i==k-1&&j>=k-1){
                        fs=Math.max(fs, max[i][j]);
                    }
                    
                }
            }
        }
        System.out.println(fs);  
    }
}