合唱团

题目描述
有 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

---

#include<iostream>
#include<limits.h>
using namespace std;

long long max(long long a,long long b) {return(a>b?a:b);}
long long min(long long a,long long b) {return(a>b?b:a);}

int main()
{
    int n,k,d;
    long long g[50]={0};//学生能力grade
    long long fmax[11][50],fmin[11][50];//相乘后数字可能太大，所以用long long
    long long fma=LLONG_MIN,fmi=LLONG_MAX;//fma用于初始化fmax数组，
    //使得fmax数组里一开始的数为long long类型的最小值，
    //这样，之后算得的值肯定比初始化的值大；fmi同理。
    long long ans=LLONG_MIN;//最终答案
    //初始化
    cin>>n;
    for(int i=0;i<n;++i)
        cin>>g[i];
    cin>>k>>d;
    for(int i=0;i<11;++i)//fmax和fmin的初始化，因为不会用memset
        for(int j=0;j<50;++j)
        {
            fmax[i][j]=fma;
            fmin[i][j]=fmi;
        }
    //f[i][j]表示选i个学生，最后选取的学生编号为j的情况。
    //fmax表示最大的情况，fmin表示最小的情况
    for(int j=0;j<n;++j)//第一轮，选取1个学生，最后选取的学生编号为j；f[1][j]就是学生本身的能力g[j]
    {
        fmax[1][j]=g[j];
        fmin[1][j]=g[j];
    }
    //思路看牛客里的，我解释不清。
    for(int i=2;i<=k;++i)
    {
        for(int j=i-1;j<n;++j)
        {
            for(int z=1;z<=d&&j-z>=i-2;++z)
            {
                fmax[i][j]=max(fmax[i][j],max((fmax[i-1][j-z]*g[j]),(fmin[i-1][j-z]*g[j])));
                fmin[i][j]=min(fmin[i][j],min((fmax[i-1][j-z]*g[j]),(fmin[i-1][j-z]*g[j])));
            }
        }
    }
    for(int i=0;i<n;++i)
        ans=max(ans,fmax[k][i]);
    cout<<ans;
    return 0;
}