合唱团

问题

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

输入描述：

每个输入包含 1 个测试用例。每个测试数据的第一行包含一个整数 n (1 <= n <= 50)，表示学生的个数，接下来的一行，包含 n 个整数，按顺序表示每个学生的能力值 ai（-50 <= ai <= 50）。接下来的一行包含两个整数，k 和 d (1 <= k <= 10, 1 <= d <= 50)。

输出描述：

输出一行表示最大的乘积。
链接：https://www.nowcoder.com/ta/2017test?query=&asc=true&order=&tagQuery=&page=1
来源：牛客网

分析：

从n 个学生中按照顺序选取 k 名学生，要求相邻两个学生的位置编号的差不超过 d： 意味着倘若我们知道最后一名学生的位置，则我们可以知道第k-1个学生的位置范围，层层递进到第一个人
使得这k 个学生的能力值的乘积最大：最优解的约束条件，且由于每个学生的能力值有正有负，所以要考虑负负得正的情况。
所以，综上所述，我们假设fm[j][i]：最后一人位置为i的选取j名学生的最大正值解，同时假设fn[j][i]：最后一人位置为i的选取j名学生的最大负值解，则有上述分析得到题解的约束方程（这里假设k-1人的位置为m）：

fm[j][i]=max(fm[j-1][m]*a[i], fn[j-1][m]*a[i])

即最后一人位置为i的选取j名学生的最大正值解=前面哥们儿位置为m（i-m<=d）的选取j-1名学生的最优值解×当前值
同理，得到fn[j][i]：

fn[j][i]=min(fm[j-1][m]*a[i], fn[j-1][m]*a[i])

即最后一人位置为i的选取j名学生的最大负值解=前面哥们儿位置为m（i-m<=d）的选取j-1名学生的最优值解×当前值
注意，由于最后一人i位置未定，所以我们需要遍历整个数组。

参考代码

#include<iostream>
#include<vector>
#include <limits.h> 
#define max(a,b)((a)>(b)?a:b)
#define min(a,b)((a)>(b)?b:a)
using namespace std;

int main()
{
 long long fm[11][51], fn[11][51];
 int n;
 cin >> n;
 vector<long long> a(n+1);
 for (int i = 1; i < n+1; i++)
 {
  cin >> a[i];
 }
 int k, d;
 cin >> k >> d;
 //vector <vector<long long>> fm(k + 1, vector<long long>(d + 1));
 //vector <vector<long long>> fn(k + 1, vector<long long>(d + 1));
 long long ans = LLONG_MIN;
 for (int i = 1; i < k+1; i++)
 {
  for (int j = 1; j < n+1; j++)
  {
   fm[i][j] = fn[i][j] = 0;
  }
 }
 for (int i = 1; i <= n; i++)
 {
  fm[1][i] = fn[1][i] = a[i];
  for (int j = 2; j <= k; j++)
  {
   for (int m = i-1;m>0&&i-m <= d; m--)
   {
    fm[j][i] = max(fm[j][i], max(fm[j - 1][m] * a[i], fn[j - 1][m] * a[i]));
    fn[j][i] = min(fn[j][i], min(fm[j - 1][m] * a[i], fn[j - 1][m] * a[i]));
   }
  }
  ans = max(ans, fm[k][i]);
 }
 cout << ans;
 return 0;
}