记忆化搜索（48双周赛）（）

 

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

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

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

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

本题要求分数和的最大值，最直接想到的就是暴力枚举所有的情况，但是如果只是这样，必然会超时。因为只是单纯的暴力枚举，会有很多重复计算，举个例子，假设元素[1,2,3,4,5,6] 如果我们第1次操作先选[1, 3]，第2次操作再选[2, 4]，剩下第3次操作[5,6]，与我们第1次操作先选[1, 2]，第2次操作选[3, 4], 剩下第3次操作也是[5, 6]这里头他们的第3次就会重复计算。因此我们可以通过记忆化来避免重复的操作
看到1 <= n <= 7，就知道这道题应该是要用状压，数组最多14个元素（占用14位），操作最多执行7次（占用3位），最多只会占用17位二进制。
在记忆化递归搜索的过程中个，相同两个元素的gcd会被计算多次，因此我们可以通过预处理来避免重复的gcd计算。

 

 

 

class Solution {
    private int[] arr;
    private int len;
    private int[] memo;
    private int[][] gcdArr;

    private int gcd(int m, int n) {
        return m % n == 0 ? n : gcd(n, m % n);
    }

    private int dp(int num, int idx) {
        if (num == 0) {
            return 0;
        }

        int key = (num << 3) + idx;
        if (memo[key] != 0) {
            return memo[key];
        }

        int ansMax = 0;
        for (int i = 0; i < len; i++) {
            if ((num & (1 << i)) != 0) {
                // 说明这一位可以选
                for (int j = 0; j < len; j++) {
                    if (j == i) {
                        continue;
                    }

                    if ((num & (1 << j)) != 0) {
                        ansMax = Math.max(ansMax, idx * gcdArr[i][j] + dp(num ^ (1 << i) ^ (1 << j), idx + 1));
                    }
                }
            }
        }

        memo[key] = ansMax;
        return ansMax;
    }

    private void calcGcd() {
        gcdArr = new int[len][len];
        for (int i = 0; i < len; i++) {
            for (int j = 0; j < len; j++) {
                gcdArr[i][j] = gcd(arr[i], arr[j]);
            }
        }
    }

    public int maxScore(int[] nums) {
        arr = nums;
        len = arr.length;
        memo = new int[1 << (len + 3)];
        calcGcd();
        return dp((1 << len) - 1, 1);
    }

}

作者：yuruiyin
链接：https://leetcode-cn.com/problems/maximize-score-after-n-operations/solution/javaji-yi-hua-sou-suo-yu-chu-li-gcd59ms-6zw66/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。