给你 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)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。