每日OJ_牛客_合唱团（打家劫舍dp）

目录

牛客_合唱团（打家劫舍dp）

解析代码1

解析代码2

---

牛客_合唱团（打家劫舍dp）

合唱团__牛客网

        有 n 个学生站成一排，每个学生有一个能力值，牛牛想从这 n 个学生中按照顺序选取 k 名学生，要求相邻两个学生的位置编号的差不超过 d，使得这 k 个学生的能力值的乘积最大，你能返回最大的乘积吗？

---

解析代码1

        题目解析：题目要求n各学生中选择k个，使这k个学生的能力值乘积最大。这是一个最优化的问题。另外，在优化过程中，提出了相邻两个学生的位置编号差不超过d的约束。 解决的方法是采用动态规划。

        代码分析：该题目是一个动态规划的问题，那么我们首先要构造出状态转移方程。 设maxVal[i][j]表示以第i个人为最后一个（前面共i个人，最后一个人必选），一共选取了j个人（包含i）时的 最大乘积。 同理，minVal[i][j]表示同样状态下的最小乘积（由于数据中存在负数，负数乘上某个极大的负数反而会变成 正的极大值，因而需要同时记录最小值）。 maxVal[i][j]很显然与maxVal[i][j-1]相关，可以理解为maxVal[i][j]由两部分组成，一部分是自身作为待选值， 另一部分是maxVal[i][j-1]加上一个人后得到的值，然后取它们的极大值，由此可以得到状态转移方程如下：

        maxVal[i][j] = max(maxVal[i][j], max(maxVal[c][j - 1] * a[i], minVal[c][j - 1] * a[i])); minVal[i][j] = min(minVal[i][j], min(maxVal[c][j - 1] * a[i], minVal[c][j - 1] * a[i])); 其中c的约束条件： i - d <= c <= i - 1 初始状态： maxVal[i][1] = a[i]; 最后遍历Maxval[i][k]即可得到最大值。

#include <iostream>
#include <vector>
#include <string>
#include <algorithm>
using namespace std;
// 数据类型必须是long long类型，不然会溢出
long long getMax(vector<int>& v, int n, int k, int d)
{
	// 状态F(i,j)： 以第i个人为最后一个人，总共选了j个人的最大值
	vector<vector<long long>> maxValue(n + 1, vector<long long>(k + 1, 0));
	vector<vector<long long>> minValue(n + 1, vector<long long>(k + 1, 0));
	// 初始化F(i, 1): 以第i个人为最后一个人，共选了1个人的最大值
	long long ret = 0;
	for (int i = 1; i <= n; ++i)
	{
		maxValue[i][1] = minValue[i][1] = v[i - 1];
	}
	for (int i = 1; i <= n; ++i)
	{
		// 需要选取k个人
		for (int j = 2; j <= k; ++j)
		{
			// 约束条件：i - d <= m <= i - 1, 且不能小于1
			for (int m = i - 1; m >= max(i - d, 1); --m)
			{
				maxValue[i][j] = max(maxValue[i][j], max(maxValue[m][j - 1] * v[i - 1],
					minValue[m][j - 1] * v[i - 1]));
				minValue[i][j] = min(minValue[i][j], min(maxValue[m][j - 1] * v[i - 1],
					minValue[m][j - 1] * v[i - 1]));
			}
		}
		// 更新最大值
		ret = max(ret, maxValue[i][k]);
	}
	return ret;
}

int main()
{
	int n, k, d;
	cin >> n;
	vector<int> v(n, 0);
	for (int i = 0; i < n; ++i)
	{
		cin >> v[i];
	}
	cin >> k;
	cin >> d;
	cout << getMax(v, n, k, d);
	return 0;
}

---

解析代码2

打家劫舍dp

#include <iostream>
#include <queue> // 里面有vector
#include <vector>
using namespace std;

#define int long long
const int INF = 0x3f3f3f3f3f3f3f3f;

signed main()
{
    int n = 0, k = 0, d = 0;
    cin >> n;
    vector<int> arr(n + 1);
    for(int i = 1; i <= n; ++i)
    {
        cin >> arr[i]; // 还有负值
    }
    cin >> k >> d;
    vector<vector<int>> f(n + 1, vector<int>(k + 1, -INF));
    vector<vector<int>> g(n + 1, vector<int>(k + 1, INF));
    // f[i][j]/g[i][j]表示从1到i挑选j个人第i个人必选的最大/小乘积
    for(int i = 1; i <= n; ++i)
    {
        f[i][1] = g[i][1] = arr[i]; // 初始化
        for(int j = 2; j <= min(i, k); ++j)
        {
            // i - prev <= d 所以 prev >= i - d
            for(int prev = max(i - d, j - 1); prev <= i - 1; ++prev) // prev代表前面挑选的最后一个位置
            {
                f[i][j] = max(f[i][j], max(f[prev][j - 1] * arr[i], g[prev][j - 1] * arr[i]));
                g[i][j] = min(g[i][j], min(f[prev][j - 1] * arr[i], g[prev][j - 1] * arr[i]));
            }
        }
    }
    int res = 0;
    for(int i = k; i <= n; ++i)
    {
        res = max(res, f[i][k]);
    }
    cout << res << endl;
    return 0;
}