【每日一题】1799. N 次操作后的最大分数和

最新推荐文章于 2024-07-22 16:27:42 发布

芊芊晚

最新推荐文章于 2024-07-22 16:27:42 发布

阅读量64

点赞数

分类专栏： LeetCode 文章标签： leetcode 深度优先算法 c++

本文链接：https://blog.csdn.net/m0_55818172/article/details/128409476

版权

LeetCode 专栏收录该内容

14 篇文章 0 订阅

订阅专栏

LeetCode 1799. N 次操作后的最大分数和

难度：困难
【Tag：状态压缩，动态规划】

题目描述

给你 nums ，它是一个大小为 2 * n 的正整数数组。你必须对这个数组执行 n 次操作。

在第 i 次操作时（操作编号从 1 开始），你需要：

选择两个元素 x 和 y 。
获得分数 i * gcd(x, y) 。
将 x 和 y 从 nums 中删除。
请你返回 n 次操作后你能获得的分数和最大为多少。

函数 gcd(x, y) 是 x 和 y 的最大公约数。

示例 1：

输入：nums = [1,2]
输出：1
解释：最优操作是：
(1 * gcd(1, 2)) = 1

示例 2：

输入：nums = [3,4,6,8]
输出：11
解释：最优操作是：
(1 * gcd(3, 6)) + (2 * gcd(4, 8)) = 3 + 8 = 11

示例 3：

输入：nums = [1,2,3,4,5,6]
输出：14
解释：最优操作是：
(1 * gcd(1, 5)) + (2 * gcd(2, 4)) + (3 * gcd(3, 6)) = 1 + 4 + 9 = 14

提示：

${1 <= n <= 7}$
${nums.length == 2 * n}$
${1 <= nums[i] <= 10^6}$

思路分析

总共有2n个数字两两配对，n的最大值为7，我们可以考虑使用状态压缩，即使用二进制来表示某一个数是否被选取。

初始状态为1111111，当有一个数字被选择的时候，我们将其改变为0。

起始状态为(1 << n) - 1依次开始搜索，我们每次选择两个数字进行配对，此时假设两个数字的位置为i 和 j，我们使用异或操作将其变为0，即mask ^ (1 << i) ^ (1 << j) ，直到我们取出所有数时，mask此时为0。

每次搜索结束时，我们保存搜索之后的最大值。

朴素的搜索会超时，所以我们要使用记忆化搜索

dp[mask]: 代表从当前状态mask出发能够获得的分数。
对于当前的状态maks来说，以前选取顺序并不影响以后能够获得的最大分数。

实现

class Solution {
public:
    int n;
    vector<int> dp;
    vector<vector<int>> gcd;
    int maxScore(vector<int>& nums) {
        n = nums.size();
        dp.resize(1 << n, 0);
        gcd.resize(n, vector<int>(n, 0)); 
        for (int i = 0; i < n; i++) {
            for (int j = i + 1; j < n; j++) gcd[i][j]= GCD(nums[i], nums[j]);
        }
        return dfs(1, (1 << n) - 1, 0, nums);
    }
    int dfs(int x, int mask, int sum, vector<int>& nums) { 
        if (x == n / 2 + 1) return sum;
        if (dp[mask] > 0) return sum + dp[mask]; // // sum为已经选取的数获得的分数 + 未选取的数能够获得的最大分数
        int ans = 0;
        for (int i = 0; i < n; i++) {
            if (((1 << i) & mask) == 0) continue;
            for (int j = i + 1; j < n; j++) {
                if (((1 << j) & mask) == 0) continue;
                int next = mask ^ (1 << i) ^ (1 << j); 
                ans = max(ans, dfs(x + 1,  next, sum + x * gcd[i][j], nums));
            }
        }
        dp[mask] = ans - sum; //ans为能够获得的总分数 减去sum后才是maks状态获得的最大分数
        return ans;
    }
    int GCD(int a, int b) {return b == 0? a : GCD(b, a % b);}
};