蓝桥杯第十题 【编程|20分】合唱团

【编程|20分】合唱团

时间限制：C/C++1秒，其他语言2秒
空间限制：C/C++32768K，其他语言65536K
64bit IO Format:%lld

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

AC代码：

#include<stdio.h>
#include<iostream>
#include<algorithm>
using namespace std;

int main()
{
	int N, K, D, i, j, k;
	long long stu[51], fm[11][51], fn[11][51], ans;
	cin >> N;
	for (i = 1; i <= N; i++)
	{
		cin >> stu[i];
	}
	cin >> K >> D;
	for (i = 0; i <= K; ++i)
	{
		for (j = 0; j <= N; ++j)
		{
			fm[i][j] = fn[i][j] = 0;
		}
	}

	for (i = 1, ans = ~0LL; i <= N; ++i)
	{
		fm[1][i] = fn[1][i] = stu[i];
		for (k = 2; k <= K; ++k)
		{
			for (j = i - 1; j > 0 && i - j <= D; --j)
			{
				fm[k][i] = max(fm[k][i], max(fm[k - 1][j] * stu[i], fn[k - 1][j] * stu[i]));
				fn[k][i] = min(fn[k][i], min(fn[k - 1][j] * stu[i], fm[k - 1][j] * stu[i]));
			}
		}
		ans = max(ans, fm[K][i]);
	}
	printf("%lld\n", ans);
	return 0;
}