1.题目描述
有 n 个学生站成一排,每个学生有一个能力值,牛牛想从这 n 个学生中按照顺序选取 k 名学生,要求相邻两个学生的位置编号的差不超过 d,使得这 k 个学生的能力值的乘积最大,你能返回最大的乘积吗?
传送门
2.code(java1.7版本)
import java.util.*;
public class Main{
public static void main(String[] args){
Scanner cin=new Scanner(System.in);
while(cin.hasNextInt()){
int n=cin.nextInt();
int[] arr=new int[n+1];
for(int i=1;i<=n;i++){
arr[i]=cin.nextInt();
}
int k=cin.nextInt();
int d=cin.nextInt();
long[][] fmax=new long[k+1][n+1];
long[][] fmin=new long[k+1][n+1];
long ans=Integer.MIN_VALUE;
for(int i=1;i<=n;i++){
fmax[1][i]=arr[i];
fmin[1][i]=arr[i];
for(int temp=2;temp<=k;temp++){
for(int j=i-1;j>=1&&i-j<=d;j--){
fmax[temp][i]=Math.max(fmax[temp][i],Math.max(fmax[temp-1][j]*arr[i],fmin[temp-1][j]*arr[i]));
fmin[temp][i]=Math.min(fmin[temp][i],Math.min(fmax[temp-1][j]*arr[i],fmin[temp-1][j]*arr[i]));
}
}
ans=Math.max(ans,fmax[k][i]);
}
System.out.println(ans);
}
}
}
3.解题思路
因为有正有负,负负得正,所以要维护两个dp数组,一个存储最大,一个存储最小。
定义fm[k][i]表示当选中了k个学生,并且以第i个学生为结尾,所产生的最大乘积;
fn[k][i]表示 当选中了k个学生,并且以第i个学生为结尾,所产生的最小乘积;
那么fm[k+1][i+1]=max(fm[k][i]*stu[i+1],fn[k][i]*stu[i+1]),
即当选中了k个学生后,再选择第i+1编号学生,所产生的最大乘积;
然而,并不能保证上一次选择的就是第i个学生,所以要遍历子数组fm[k],
令j从i到1,并且j与i+1之间小于间隔D,遍历fm[k][j],以及fn[k][j];
同理fn[k+1][i+1]=min(fn[k][i]*stu[i+1],fm[k][i]*stu[i+1])。
最后,遍历一遍fm[K][i]求得最大值(i从1~N)。