网易校招-合唱团

题目描述
有 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

题目要求比较简单，也很容易想到用动态规划来做，依次找出1个人，2个人的最大值，以此到k个人的最大值。但是这题有个坑，那就是能力值可能是负数（好吧，我以为战五渣已经够低了，没想到还有是负数的）
代码和解析如下。

#include "stdafx.h"
#include <iostream>
#include <algorithm>
#include <vector>
#define maxn 55 
using namespace std;
//因为可能存在负数，所以用了两个数组，一个储存的当前位置的最大值，一个储存最小值
//状态转移方程如下
//dpmax[i][j] = max(dpmax[i][j], max(dpmax[r][j - 1] * arr[i], dpmin[r][j - 1] * arr[i]));

long long dpmax[maxn][15]; //其中一维记录当前位置，二维记录总人数

long long dpmin[maxn][15];
inline long long  max(long  long a, long long b) {
    if (a > b)
        return a;
    return b;
}
inline long long  min(long long a, long long b) {
    if (a < b)
        return a;
    return b;
}
int main() {
    int n, k, d;
    cin >> n;
    int *arr = new int[n];
    for (int i = 0; i < n; ++i)
        cin >> arr[i];
    cin >> k >> d;
    //初始化
    for (int i = 0; i < n; ++i) {
        dpmax[i][1] = arr[i];
        dpmin[i][1] = arr[i];
    }
    for (int i = 0; i < n; ++i)
        for (int j = 2; j <= k; ++j)
            for (int r = max(0, i - d); r <= i - 1; ++r) {//这里注意判断条件d
                dpmax[i][j] = max(dpmax[i][j], max(dpmax[r][j - 1] * arr[i], dpmin[r][j - 1] * arr[i]));
                dpmin[i][j] = min(dpmin[i][j], min(dpmin[r][j - 1] * arr[i], dpmax[r][j - 1] * arr[i]));
            }
    long long  res = dpmax[0][k];
    for (int i = 0; i < n; ++i)
        res = max(res, dpmax[i][k]);
    cout << res << endl;
}