有 n 个学生站成一排,每个学生有一个能力值,牛牛想从这 n 个学生中按照顺序选取 k 名学生,要求相邻两个学生的位置编号的差不超过 d,使得这 k 个学生的能力值的乘积最大,你能返回最大的乘积吗?
输入描述:
每个输入包含 1 个测试用例。每个测试数据的第一行包含一个整数 n (1 <= n <= 50),表示学生的个数,接下来的一行,包含 n 个整数,按顺序表示每个学生的能力值 ai(-50 <= ai <= 50)。接下来的一行包含两个整数,k 和 d (1 <= k <= 10, 1 <= d <= 50)。
输出描述:
输出一行表示最大的乘积。
输入例子:
3 7 4 7 2 50
输出例子:
49
动态规划:考虑正负,需保存乘积最大值和乘积最小值。 dp0[i][j]表示选中第i个数长度为j的乘积最大, dp1[i][j]表示选中第i个数长度为j的乘积最小。 长度为j的乘积由长度为j-1转化而来,即dp[m][j-1]*a[i] m的取值为[max(1,i-d),i-1]。
#include <iostream> using namespace std; typedef long long ll; ll dp0[51][11]; ll dp1[51][11]; int a[51]; ll ans; int main() { int n,k,d; cin>>n; for(int i=1;i<=n;i++){ cin>>a[i]; } cin>>k>>d; for(int i=1;i<=n;i++){ dp0[i][1]=a[i]; dp1[i][1]=a[i]; for(int j=2;j<=k;j++){ for(int m=max(1,i-d);m<i;m++){ dp0[i][j]=max(dp0[i][j],max(a[i]*dp0[m][j-1],a[i]*dp1[m][j-1])); dp1[i][j]=min(dp1[i][j],min(a[i]*dp1[m][j-1],a[i]*dp0[m][j-1])); } } ans=max(ans,dp0[i][k]); cout<<ans<<endl; } cout<<ans; return 0; }
参考:http://www.nowcoder.com/test/question/done?tid=4826956&qid=45830