Leetcode 1770：执行乘法运算的最大分数

题目描述

给你两个长度分别 n 和 m 的整数数组 nums 和 multipliers ，其中 n >= m ，数组下标 从 1 开始 计数。

初始时，你的分数为 0 。你需要执行恰好 m 步操作。在第 i 步操作（从 1 开始 计数）中，需要：

选择数组 nums 开头处或者末尾处 的整数 x 。
你获得 multipliers[i] * x 分，并累加到你的分数中。
将 x 从数组 nums 中移除。
在执行 m 步操作后，返回 最大 分数。

 

示例 1：

输入：nums = [1,2,3], multipliers = [3,2,1]
输出：14
解释：一种最优解决方案如下：
- 选择末尾处的整数 3 ，[1,2,3] ，得 3 * 3 = 9 分，累加到分数中。
- 选择末尾处的整数 2 ，[1,2] ，得 2 * 2 = 4 分，累加到分数中。
- 选择末尾处的整数 1 ，[1] ，得 1 * 1 = 1 分，累加到分数中。
总分数为 9 + 4 + 1 = 14 。
示例 2：

输入：nums = [-5,-3,-3,-2,7,1], multipliers = [-10,-5,3,4,6]
输出：102
解释：一种最优解决方案如下：
- 选择开头处的整数 -5 ，[-5,-3,-3,-2,7,1] ，得 -5 * -10 = 50 分，累加到分数中。
- 选择开头处的整数 -3 ，[-3,-3,-2,7,1] ，得 -3 * -5 = 15 分，累加到分数中。
- 选择开头处的整数 -3 ，[-3,-2,7,1] ，得 -3 * 3 = -9 分，累加到分数中。
- 选择末尾处的整数 1 ，[-2,7,1] ，得 1 * 4 = 4 分，累加到分数中。
- 选择末尾处的整数 7 ，[-2,7] ，得 7 * 6 = 42 分，累加到分数中。
总分数为 50 + 15 - 9 + 4 + 42 = 102 。
 

提示：

n == nums.length
m == multipliers.length
1 <= m <= 103
m <= n <= 105
-1000 <= nums[i], multipliers[i] <= 1000

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/maximum-score-from-performing-multiplication-operations
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

 

 

解题思路

第一反应是dfs，但是会超时

class Solution {
public:
    int ans = 0;
    void dfs(vector<int> &nums, int l, int r, vector<int> &mul, int idx, int cur){
        if(idx >= mul.size()){
            ans = max(ans, cur);
            return;
        }
        dfs(nums, l+1, r, mul, idx+1, cur + nums[l] * mul[idx]);
        dfs(nums, l, r-1, mul, idx+1, cur + nums[r] * mul[idx]);
    }
    int maximumScore(vector<int>& nums, vector<int>& multipliers) {
        dfs(nums, 0, nums.size()-1, multipliers, 0, 0);
        return ans;
    }
};

 

动态规划

dp[i][j] 表示前面取i个，后边取j个，i+j=k

class Solution {
public:
    int maximumScore(vector<int>& nums, vector<int>& multipliers) {
        int m = multipliers.size();
        int n = nums.size();
        vector<vector<int>> dp(m+1, vector<int>(m+1));
        for(int k = 1; k <= m; k++){
            for(int i = 0; i <= k; i++){
                if(i == 0) dp[i][k-i] = dp[i][k-i-1] + nums[n-k+i]*multipliers[k-1];
                else if(i == k) dp[i][k-i] = dp[i-1][k-i] + nums[i-1] * multipliers[k-1];
                else{
                    int head = dp[i][k-i-1] + nums[n-k+i]*multipliers[k-1];
                    int tail = dp[i-1][k-i] + nums[i-1] * multipliers[k-1];
                    dp[i][k-i] = max(head, tail);
                }
            }
        }
        int ans = INT_MIN;
        for(int i = 0; i <= m; i++) ans = max(ans, dp[i][m-i]);
        return ans;
    }
};