给出一个由无重复的正整数组成的集合, 找出其中最大的整除子集, 子集中任意一对 (Si, Sj) 都要满足: Si % Sj = 0 或 Sj % Si = 0。
如果有多个目标子集,返回其中任何一个均可。
示例 1:
集合: [1,2,3] 结果: [1,2] (当然, [1,3] 也正确)
示例 2:
集合: [1,2,4,8] 结果: [1,2,4,8]
思路:这道题先对数组由大到小排序,如果满足nums[i]%nums[i+1]==0,那么nums[i+1]一定可以被能整除nums[i]的所有元素整除,所以如果满足nums[i]%nums[i+1]==0并且dp[i]+1>=dp[i+1](dp是我们维护的递推数组,dp[i]表示到第i个元素最大的子集的个数),就更新dp[i+1]=dp[i]+1,递推公式就这么简单,但是本文需要求出集合的元素,而不仅仅是个数。所以我们还需要记录哪个dp[i]是最大的并且记录dp[i]对应子集是什么。
对于记录哪个dp[i]最大,可以用一个global_max来记录即可,但是如何记录每个dp[i]的子集呢,我们需要另外一个parent递推数组,如果满足nums[i]%nums[i+1]==0并且dp[i]+1>=dp[i+1]的条件,我们就更新parent[i+1]=i,parent[k]=q的意思是记录第k个元素的最大子集开始的下标是q,有了开始下标就可以利用回溯的思想进行返回查找,比如我们最后记录到dp[15]有最大的子集,并且dp[15]=4,意味着有4个子集,那我们利用如下公式回溯4个子集:
for (int i = 0; i < mx; i++) {
res.push_back(nums[mx_index]);
mx_index = parent[mx_index];
}
mx表示dp[15]=4,首先放入15的下标,即mx_index=15,然后把mx_index更新为递推数组parent中存的以mx_index为下标的子集,即parent[mx_index],由于我们初始化parent[i]=i,即回溯自身,那么到最后一个子集(只包含一个元素),一定就是自身。
参考代码:
class Solution {
public:
vector<int> largestDivisibleSubset(vector<int>& nums) {
vector<int> res;
if (nums.size() == 0) {
return {};
}
int *dp = new int[nums.size()];
for(int i=0;i<nums.size();i++){
dp[i]=1;
}
int *parent = new int[nums.size()];
memset(parent, 0, nums.size() * sizeof(int));
int mx = 0;
int mx_index = 0;
sort(nums.begin(), nums.end(), greater<int>());
for (int i = 1; i < nums.size(); i++) {
for (int j = 0; j < i; j++) {
if (nums[j] % nums[i] == 0) {
if (dp[i] <= (dp[j] + 1)) {
dp[i] = dp[j] + 1;
parent[i] = j;
if (dp[i] >= mx) {
mx = dp[i];
mx_index = i;
}
}
}
}
}
for (int i = 0; i < mx; i++) {
res.push_back(nums[mx_index]);
mx_index = parent[mx_index];
}
if (res.size() == 0) {
res.push_back(nums[0]);
}
delete[] dp;
delete[] parent;
return res;
}
};