2016网易笔试编程题

题目描述

有 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

题解：这个题一开始想到的是的递归搜索+剪枝，然后发现在递归的过程中有很多的值是重复计算的，而且递归的复杂度太高，然后转到dp思路上来。笔者的dp确是很菜，以前做的关于dp的算法也是比较少的。

言归正传，将题目抽象出来就是从n个数中选择k个数通过某种运算得到极值（无非就是极大值或者极小值），以前做过类似的题目，确实用dp做的。dp[i][j]表示以a[i]结尾选择j个数后的极值。这样的就好处理了。题目中有两个条件：1）有负数存在，2）相邻的数的index最大是d。

    先处理第一个条件，有负数的话就有可能偶数个负数相乘的来的。那么假设a[i]是负数的话，要想dp[i][j]最大，那么dp[k][j-1]应该取最小的才行（当然最好是负数了）。所以就保存两个dp数组，一个存最大，一个存最小。状态转移方程可以直接看代码。

    第二个条件就更好处理了，确定了i后，往前推d个数就行了，也就是在i-d~i-1之间的数来转移方程。

#include<bits/stdc++.h>
#define LL long long
#define INF 1e17
using namespace std;

LL mx[51][51],mi[51][51];
LL a[51];
int main(){
    int k,d,n;
    scanf("%d",&n);
    for(int i = 1;i<=n;i++)cin>>a[i];
    for(int i = 1;i<=n;i++)for(int j = 1;j<=n;j++)mx[i][j] = -INF,mi[i][j] = INF;
    for(int i = 1;i<=n;i++)mx[i][1] = mi[i][1] = a[i];
    cin>>k>>d;
    for(int i = 2;i<=n;i++){
        for(int j = 2;j<=min(i,n);j++){
            for(int h = i-1;h>=max(1,i-d)&&h>=j-1;h--){
                mx[i][j] = max(mx[i][j],max(mx[h][j-1]*a[i],mi[h][j-1]*a[i]));
                mi[i][j] = min(mi[i][j],min(mi[h][j-1]*a[i],mx[h][j-1]*a[i]));
            }
        }
    }
    LL ans = -INF;
    for(int i = 1;i<=n;i++)ans = max(ans,mx[i][k]);
    cout<<ans<<endl;
    return 0;
}