给你 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] <= 106
这题相当于把2n个人分为n组,问有多少种不同的分法。
一种显然但错误的思路是先选2个,再在剩下的2n-2个里面选2个,以此类推,选法一共是
C2n2∗C2n−22∗...∗C22=(2n)!/2n
事实上第一次选出(1,2),第二次选出(3,4)和第一次选出(3,4),第二次选出(1,2)是一样的分法,因此这里有重复,最终还需要除以组数的全排列数,即总分法为
C2n2∗C2n−22∗...∗C22=(2n)!/(2n∗n!)
带入n=7算出运行次数为135135,乘上计算答案时需要排序用到O(nlogn)的复杂度,运行次数再乘以10大约为1e6,说明用暴力回溯的方法是可以通过的。 用回溯的方法计算出每个组合后,将每个组合内的最大公因数按从小到大排序然后按顺序计算答案即可。 为什么排序后能保证答案最大呢?这是因为根据排序不等式
a1∗b1+a2∗b2+...+an∗bn
当a和b的大小顺序保持一致时达到最大。 这里操作次数i是递增的,因此对得到的n个最大公因数进行排序就能得到最大值了。 代码如下:
class Solution {
public:
int ans = 0;
vector<int>tmp;
void dfs(vector<int>& nums,vector<int>& a,vector<bool>& v,int pos,int m){
if(pos==nums.size()){
int res = 0;
auto r = tmp;
sort(r.begin(),r.end());//排序后计算答案
for(int i=0;i<a.size();++i)res += (i+1)*r[i];
ans = max(ans,res);
return;
}
if(pos&1){//奇数位置计算最大公因数
for(int i=m;i<nums.size();++i){
if(v[i])continue;
tmp[pos/2] = __gcd(a[pos/2],nums[i]);v[i] = 1;
if(i==m)dfs(nums,a,v,pos+1,m+1);
else dfs(nums,a,v,pos+1,m);
v[i] = 0;//回溯
}
}
else{//偶数位置取未使用的最小值
while(v[m])m++;
a[pos/2] = nums[m];
dfs(nums,a,v,pos+1,m+1);
}
}
int maxScore(vector<int>& nums) noexcept{
vector<int>a(nums.size()/2);
vector<bool>v(nums.size());
tmp = a;
dfs(nums,a,v,0,0);
return ans;
}
};