合唱团

/***************合唱团
有 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
******************************/

/**************基本思路
动态规划：(感觉规划地不是很好，虽然最终结果通过了，但是debug过程很艰辛。。。。)
ak > 0, Fmax(k,i)=max{Fmax(k-1,i-dd)*a(i)},    for(1<= dd <= d)       这里k表示当前选了几个人，i表示当前这个人的位置。 dd是偏差值
ak < 0, Fmax(k,i)=max{Fmin(k-1,i-dd)*a(i)},                          
******************************/

#include<iostream>
#include<vector>
using namespace std;

long long int choir(int n, int k, int d, int ai[] )   //注意数据类型long long int
{
    long long int f_max[k][n];
    long long int f_min[k][n];
    for(int i=0;i<n;i++)
    {
        f_max[0][i] = ai[i];
        f_min[0][i] = ai[i];
    }
    for(int i=1;i<k;i++)      //初始化
    {
        for(int j=0;j<n;j++)
        {
            f_max[i][j] = 1;
            f_min[i][j] = 1;
        }
    }
    for(int a=1; a<k; a++)
    {
        for(int b=a; b<n; b++)
        {
            long long int temp_max;
            long long int temp_min;
            temp_max = f_max[a-1][b-1] * ai[b];    //赋初值要注意
            temp_min = f_max[a-1][b-1] * ai[b];   //这里也要乘mx不能乘min
            for(int dd = 1; dd<=d ;dd++)         //必须要从dd=1开始， 如果从2开始，初值会错误，不进入负数判断
            {
                if(b-dd>=a-1)
                {
                    temp_min = min(temp_min,f_max[a-1][b-dd] * ai[b]);
                    if(ai[b]>=0)
                        temp_max = max(temp_max,f_max[a-1][b-dd] * ai[b]);
                    else
                        temp_max = max(temp_max,f_min[a-1][b-dd] * ai[b]);
                }
            }
            f_max[a][b] = temp_max;
            f_min[a][b] = temp_min;
        }
    }
    long long int max_result = 1;
    for(int a=0;a<k;a++)
        max_result *= ai[a];
    for(int b=k-1; b<n; b++)
    {
        max_result = max(max_result,f_max[k-1][b]);
    }
    return max_result;
}
int main()
{
    int n,k,d;
    cin>>n;
    int ai[n];
    for(int i=0; i<n; i++)
        cin>>ai[i];
    cin>>k>>d;
    cout<<choir(n, k, d, ai);
    return 0;

}