校招编程——合唱团

题目描述

有 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<vector>
inline long long max(long long a, long long b) {return a>b?a:b;}
inline long long min(long long a, long long b) {return a<b?a:b;}
using namespace std;

int main(){
    int n;
    cin >> n;
    vector<int> a;
    for (int i=0; i<n; i++){
        int ai;
        cin >> ai;
        a.push_back(ai);
    }
    int k, d;
    cin >> k >> d;
    vector<vector<long long>> fmax(k, vector<long long> (n, 0));
    vector<vector<long long>> fmin(k, vector<long long> (n, 0));
    
    long long res=0;
    
    for(int i=0; i<n; i++){
        fmax[0][i] = fmin[0][i] = a[i];
        for(int kk=1; kk<k; kk++){
            for(int j=i-1; j>=0 && i-j<=d; j--){
                fmax[kk][i] = max(fmax[kk][i], max(fmax[kk-1][j]*a[i],
                                 fmin[kk-1][j]*a[i]));
                fmin[kk][i] = min(fmin[kk][i], min(fmax[kk-1][j]*a[i],
                                 fmin[kk-1][j]*a[i]));
            }
        }
        res = max(res, fmax[k-1][i]);
    }
    cout << res;
    return 0;
}