【牛客网】合唱团

题目描述

有 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

f[i][j][0/1]表示选中第i个人后已经有j个人，0或1表示最大或最小乘积（要考虑到负数）

#include
#include
using namespace std;
long long f[51][51][2];
int a[55];
int main(){
	int n,k,d;
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
	scanf("%d",&a[i]);
	scanf("%d%d",&k,&d); 
	long long ans=0;
	for(int i=1;i<=n;i++){//遍历每一个人人
		f[i][1][0]=f[i][1][1]=a[i];//仅选一个人时，乘积为本身
		for(int j=2;j<=k;j++){
			for(int l=i-1;l>=max(i-d,1);--l){
			//在第i个数之前取一个与目前数乘积最大的 ，且与当前数距离不超过d 
				f[i][j][0]=max(f[i][j][0],max(f[l][j-1][1]*a[i],f[l][j-1][0]*a[i]));
				f[i][j][1]=min(f[i][j][1],min(f[l][j-1][1]*a[i],f[l][j-1][0]*a[i]));
			}	
		}
		ans=max(ans,max(f[i][k][0],f[i][k][1]));
	}
	printf("%lld\n",ans);
	return 0;
}