-题目
Given a set of distinct positive integers, find the largest subset such that every pair (Si, Sj) of elements in this subset satisfies: Si % Sj = 0 or Sj % Si = 0.
If there are multiple solutions, return any subset is fine.
Examples:
nums: [1,2,3]
Result: [1,2] (of course, [1,3] will also be ok)
nums: [1,2,4,8]
Result: [1,2,4,8]
翻译:
题目的意思是求一个数列的最大子集,这个子集中的元素两两可以互相整除。
-思路
这道题的解题思路是用一个数组来记录某个元素加入子集以后能够获得的最大长度,也即当元素 i能够整除某个元素j时,它加入子集能够获得的最大长度的动规转移方程为:max_sum[i] = max(max_sum[i],max_sum[j]+1),然后为了节省空间我们用一个previous数组来存储元素i加入子集以后前面最大的能整除元素i一个元素,我们用max_index来记录最长长度的子集的最后一个元素,然后再通过一个遍历来生成最长子集。
-代码
class Solution {
public:
vector<int> largestDivisibleSubset(vector<int>& nums) {
vector<int> subset;
if(nums.size() == 0) return subset;
sort(nums.begin(), nums.end());
vector<int> previous(nums.size()+1, 0);
vector<int> max_sum(nums.size()+1, 1);
int max_index = 1, max_max = 0;
for(int i = 1; i <= nums.size(); i++) {
for(int j = i - 1; j >= 1; j--) {
if(nums[i-1] % nums[j-1] == 0) {
if(max_sum[j]+1 > max_sum[i]) { //如果前序的集合个数较大则替换前序
max_sum[i] = max_sum[j] + 1;
previous[i] = j;
cout << i << " " << j << endl;
if(max_sum[i] > max_max) {
max_max = max_sum[i];
max_index = i;
}
}
}
}
}
while(max_index != 0) {
subset.push_back(nums[max_index-1]);
max_index = previous[max_index];
}
return subset;
}
};