合唱团 牛客网动态规划题目

题目链接

题目采用动态规划的方式解决

动态表示：g[i][j] 从a[1…i]中选择j个人，且最后一个人必须选，能力的最大乘积
f[i][j] 表示从a[1…i]中选择j个人，且最后一个人必须选，能力的最小乘积

解释：为什么最后一个人必须选，因为选择了最后一个人，就可根据d判断可以选择距离最远的哪个人
为什么还要一个数组表示最小乘积呢？因为当a[i]为负数时，越小的数字乘以负数更大，所以还要一个数组表示最小乘积

动态转移方程：prew表示要往前选择的人，i - prew <= d，因为题目限制相邻两个人的最远距离，所以prew >= i - d，并且要往前选择j-1个人，所以prew > j - 1，总结：prew = max(i-d,j-1)
*g[i][j] = max(max(g[prew][j-1]*a[i],f[prew][j-1]a[i]),g[i][j])
prew从j-1往前直到超出范围，注意a[i]不能提出来，因为a[i]可能是负数
f[i][j] = min(min(g[prew][j-1]*a[i],f[prew][j-1]*a[i]),g[i][j])

初始化：由于f[prew][j]表示从a[1…prew]选择j个人，所以prew大于等于j，所以表的右上方不能用，没有意义
由于prew是变化的，填表的值要前一列，所以要初始化第一列，第一列表示只能挑选一个人，并且a[i]必须挑选，所以g[i][0] = f[i][0] = a[i]
由于在初始化表格的时候，要比较g[i][j]，因为他是变化的，但是在第一次初始化的时候，g[i][j]如果初始化为0的话，当乘积为负数的时候，却没办法更新，因为最大值为0，所以这里就不能初始为0，初始化为INT_MIN、INT_MAX，当然不行，因为数据的范围是非常大的，-50，如果挑选10个，(-50)^10，所以数据类型应该为long long，分别更新为0x3f3f3f3f3f3f3f3f和-0x3f3f3f3f3f3f3f3f

填表顺序：从上到下，从左往右

返回值：max(g[k][k],…,g[n][k])，表示从前k个人选择k个人，到从n个人选择k个人的最大值

#include <iostream>
#include <vector>
using namespace std;

int main() 
{
    int n = 0;
    cin >> n;
    
    vector<int> a(n+1,0);
    for(int i = 1; i < n + 1; ++i)//从1开始输入，方便后续更新动态数组
    {
        cin >> a[i];
    }

    int k,d;
    cin >> k >> d;

    vector<vector<long long>> g(n+1,vector<long long>(k+1,-0x3f3f3f3f3f3f3f3f));//n个人，选择k个人，能力的最大乘积
    vector<vector<long long>> f(n+1,vector<long long>(k+1,0x3f3f3f3f3f3f3f3f));//n个人，选择k个人，能力的最小乘积
    
    for(int i = 1; i <= n; ++i)//初始化第二列，即只挑选一人，并且最后一人a[i]必须被选择
    {
        g[i][1] = f[i][1] = a[i];
    }

    for(int j = 2; j < k + 1; ++j)
    {
        for(int i = j; i < n + 1; ++i)//注意要在表格的左下角，包括右对角线
        {
            int prew = max(j-1,i-d);
            while(prew <= i - 1)
            {
                g[i][j] = max(max(g[prew][j-1]*a[i],f[prew][j-1]*a[i]),g[i][j]);
                f[i][j] = min(min(g[prew][j-1]*a[i],f[prew][j-1]*a[i]),f[i][j]);
                ++prew;
            }
        }
    }

    long long result = -0x3f3f3f3f3f3f3f3f;
    for(int i = k; i < n + 1; ++i)//挑选k个人，所以肯定要从第k个人挑，不然前面哪有那么多人给你挑
    {
        result = max(g[i][k],result);
    }

    cout << result << endl;
    return 0;
}