题目描述
有 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<cstdio> #include<algorithm> 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; }